Small fix in the run task tests