corrected kfold class name