Fixed missed entry in training set bug