fixed split_train_test bug