Forgot to update the tests :P