Fix tests not using the actual task.