Forgot to change the tests...