Fix calls to tasks into tests