fixing minor big in call_train_script