Fixed tests

I forgot to commit this file sorry :/