Source code fix

messed up with git, didn't pull before I pushed :S