Fix task tests