small bug in training_IDs found