fixed training set change