FIX : test due to KFold change
FIX : better fix of KFold balance