Updated task tests