Adapted tests for loop flat task.