diff --git a/train_for_multidataset.py b/train_for_multidataset.py
index 0e08815..9ace336 100644
--- a/train_for_multidataset.py
+++ b/train_for_multidataset.py
@@ -141,7 +141,7 @@ def MultiObsToEnv(multi_data_url, workroot):
                 multi_data_json[i]["dataset_url"], path) + str(e))
     return 
 ### Copy the output model to obs ###  
-def EnvToObs(obs_train_url, train_dir):
+def EnvToObs(train_dir, obs_train_url):
     try:
         mox.file.copy_parallel(train_dir, obs_train_url)
         print("Successfully Upload {} to {}".format(train_dir,