Fix the task helper - not sure why this isn't caught by the spec for the mock.