* Fix a training set too small bug