Fix Kfold iterator so it returns test set indexes.