diff --git a/llvm/utils/lit/pyproject.toml b/llvm/utils/lit/pyproject.toml
new file mode 100644
--- /dev/null
+++ b/llvm/utils/lit/pyproject.toml
@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
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
@@ -1,5 +1,5 @@
-import lit
 import os
+import sys
 
 from setuptools import setup, find_packages
 
@@ -8,6 +8,9 @@
 #   python path/to/setup.py install
 # to work (for scripts, etc.)
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
+sys.path.insert(0, ".")
+
+import lit
 
 with open("README.rst", "r", encoding="utf-8") as f:
     long_description = f.read()