training set bug fix