fixed a bug in five fold data split