diff --git a/setup.py b/setup.py
index 97f6623..e2edf38 100644
--- a/setup.py
+++ b/setup.py
@@ -7,9 +7,7 @@
 from setuptools.command.build_py import build_py
 
 # TODO: find from extras maybe
-HOST_MLIR_PYTHON_PACKAGE_PREFIX = os.environ.get(
-    "HOST_MLIR_PYTHON_PACKAGE_PREFIX", "mlir"
-)
+HOST_MLIR_PYTHON_PACKAGE_PREFIX = "triton_mlir"
 PACKAGE_NAME = f"{HOST_MLIR_PYTHON_PACKAGE_PREFIX.replace('.', '-').replace('_', '-')}-python-extras"