Fixes #2

I hade same problem as ~nooone so I just fixed it :)