diff --git a/python/setup.py b/python/setup.py
index 05f008d2c..d8ddb327c 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -140,6 +140,7 @@ setup(
         "cmake",
         "filelock",
         "torch",
+        "lit",
     ],
     package_data={"triton/ops": ["*.c"], "triton/ops/blocksparse": ["*.c"]},
     include_package_data=True,