diff --git a/llvm/utils/lit/lit/__init__.py b/llvm/utils/lit/lit/__init__.py
index d364464d7789..6bf0132d4252 100644
--- a/llvm/utils/lit/lit/__init__.py
+++ b/llvm/utils/lit/lit/__init__.py
@@ -6,4 +6,3 @@ __versioninfo__ = (0, 11, 0)
 __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
index 9d26b2ed71c7..9224ec146758 100644
--- a/llvm/utils/lit/setup.py
+++ b/llvm/utils/lit/setup.py
@@ -71,7 +71,7 @@ https://github.com/llvm/llvm-project/tree/master/llvm/utils/lit
     packages = find_packages(),
     entry_points = {
         'console_scripts': [
-            'lit = lit:main',
+            'lit = lit.main:main',
             ],
         }
 )