Forgot to update the method signature in test cases