Fixed tests

Freaking tests were not written correctly…