diff --git a/llvm/utils/lit/lit/__init__.py b/llvm/utils/lit/lit/__init__.py
--- a/llvm/utils/lit/lit/__init__.py
+++ b/llvm/utils/lit/lit/__init__.py
@@ -6,4 +6,3 @@
 __version__ = '.'.join(str(v) for v in __versioninfo__) + 'dev'
 
 __all__ = []
-from .main import main
diff --git a/llvm/utils/lit/setup.py b/llvm/utils/lit/setup.py
--- a/llvm/utils/lit/setup.py
+++ b/llvm/utils/lit/setup.py
@@ -71,7 +71,7 @@
     packages = find_packages(),
     entry_points = {
         'console_scripts': [
-            'lit = lit:main',
+            'lit = lit.main:main',
             ],
         }
 )