fixed build training set