diff --git a/import.py b/import.py
index 5819462c..683ef03c 100644
--- a/import.py
+++ b/import.py
@@ -1,9 +1,16 @@
+"""
+python import.py export.zip challenges,teams,both,metadata
+"""
 from CTFd import create_app
 from CTFd.utils import import_ctf
 
-import zipfile
 import sys
 
 app = create_app()
 with app.app_context():
-    import_ctf(sys.argv[1])
+    if sys.argv[2]:
+        segments = sys.argv[2].split(',')
+    else:
+        segments = None
+
+    import_ctf(sys.argv[1], segments=segments)