fixing training set