Forgot to update the tests...