Fixing task tests