Bugfix

Appeared when SVM with balance = 1