2 files changed, 3 insertions, 9 deletions
diff --git a/llvm/utils/lit/lit.py b/llvm/utils/lit/lit.py
index af070c69b5b..2f1fb43b11b 100755
--- a/llvm/utils/lit/lit.py
+++ b/llvm/utils/lit/lit.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
-import sys
 
-from lit.main import main
+import lit.main
 
-if __name__=='__main__':
-    main()
+if __name__ == '__main__':
+    lit.main()
diff --git a/llvm/utils/lit/lit/main.py b/llvm/utils/lit/lit/main.py
index e56cd44dd40..7e8db087932 100755
--- a/llvm/utils/lit/lit/main.py
+++ b/llvm/utils/lit/lit/main.py
@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-
 """
 lit - LLVM Integrated Tester.
 
@@ -381,6 +379,3 @@ def write_test_results_xunit(tests, opts):
         xunit_output_file.write("</testsuite>\n")
     xunit_output_file.write("</testsuites>")
     xunit_output_file.close()
-
-if __name__=='__main__':
-    main()