diff --git a/create_syn_data.sh b/create_syn_data.sh
index 35573b2..5904d47 100755
--- a/create_syn_data.sh
+++ b/create_syn_data.sh
@@ -4,4 +4,4 @@ cd data/lcn
 python setup.py build_ext --inplace
 
 cd ../
-python create_syn_data.py $2
+python create_syn_data.py $1
diff --git a/data/create_syn_data.py b/data/create_syn_data.py
index 809206e..2615235 100644
--- a/data/create_syn_data.py
+++ b/data/create_syn_data.py
@@ -228,10 +228,10 @@ if __name__ == '__main__':
     out_root.mkdir(parents=True, exist_ok=True)
 
     start = 0
-    if len(sys.argv) >= 2 and isinstance(sys.argv[2], int):
-        start = sys.argv[2]
+    if len(sys.argv) >= 2 and isinstance(sys.argv[1], int):
+        start = sys.argv[1]
     else:
-        if sys.argv[2] == '--resume':
+        if sys.argv[1] == '--resume':
             try:
                 start = max([int(dir) for dir in os.listdir(out_root) if str.isdigit(dir)]) or 0
             except: