fix cross-validation return