Fix some failing task tests