diff --git a/nvfp4_utils.py b/nvfp4_utils.py
index b95012e9c..244b372fb 100644
--- a/nvfp4_utils.py
+++ b/nvfp4_utils.py
@@ -4,7 +4,7 @@
 
 import triton
 import triton.language as tl
-from triton_kernels.target_info import cuda_capability_geq
+from triton.language.target_info import cuda_capability_geq
 
 MXFP_BLOCK_SIZE = tl.constexpr(16)