Forgot to change one method signature.