Correct TrainCV handling