Fixed printing bug in training