From e260369be1b6548ff97456c43c952357949dde7c Mon Sep 17 00:00:00 2001
From: Brian Warner <warner@lothar.com>
Date: Tue, 10 Apr 2018 21:33:50 -0700
Subject: [PATCH] __main__.py: stop breaking automat-visualize

---
 src/wormhole/__main__.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

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