diff --git a/setupbase.py b/setupbase.py
index 585dc43..f447b05 100644
--- a/setupbase.py
+++ b/setupbase.py
@@ -336,8 +336,8 @@ def find_entry_points():
 
 script_src = """#!{executable}
 # This script was automatically generated by setup.py
-from {mod} import {func}
 if __name__ == '__main__':
+    from {mod} import {func}
     {func}()
 """