diff --git a/llvm/utils/lit/lit.py b/llvm/utils/lit/lit.py
index 851063b..2c5ecfe 100755
--- a/llvm/utils/lit/lit.py
+++ b/llvm/utils/lit/lit.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 
+from lit.main import main
+
 if __name__=='__main__':
-    import lit
-    lit.main()
+    main()