Forgot to change the tests.