Fix #1

Dat fucking ;
But now there is another bug.
I suck at coding