Fixed tests
Forgot this