Fixed some errors in TrainTests