Oops. Forgot to fix the tests.