diff --git a/import.py b/import.py
index 7dbcb647..3671820b 100644
--- a/import.py
+++ b/import.py
@@ -8,7 +8,7 @@ import sys
 
 app = create_app()
 with app.app_context():
-    if len(sys.argv) == 2:
+    if len(sys.argv) == 3:
         segments = sys.argv[2].split(',')
     else:
         segments = None