change cluster from train_test to train