corrected error in label_train