Oops. Forgot to update the tests after the last commit