Fix wrong mock function.