Fixed bug (thanks Klaas =D)