diff --git a/setup.py b/setup.py
index e330293..dae7e33 100644
--- a/setup.py
+++ b/setup.py
@@ -4,9 +4,7 @@
 from setuptools import setup, find_namespace_packages
 
 # 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"