diff --git a/src/wormhole/__main__.py b/src/wormhole/__main__.py
index d6c1b97..37e97b7 100644
--- a/src/wormhole/__main__.py
+++ b/src/wormhole/__main__.py
@@ -1,7 +1,7 @@
 from __future__ import absolute_import, print_function, unicode_literals
-from .cli import cli
-
-if __name__ != "__main__":
-    raise ImportError('this module should not be imported')
-
-cli.wormhole()
+if __name__ == "__main__":
+    from .cli import cli
+    cli.wormhole()
+else:
+    # raise ImportError('this module should not be imported')
+    pass