Fixed tests

Yay