If-statements fix!

Stupid me added the wrong if-statements. They are now correct.