diff --git a/python/sgl_kernel_npu/sgl_kernel_npu/fla/l2norm.py b/python/sgl_kernel_npu/sgl_kernel_npu/fla/l2norm.py
index 7c52c70ee..3f41d2e59 100644
--- a/python/sgl_kernel_npu/sgl_kernel_npu/fla/l2norm.py
+++ b/python/sgl_kernel_npu/sgl_kernel_npu/fla/l2norm.py
@@ -42,10 +42,10 @@ def l2norm_fwd_kernel_opt(
     x,
     y,
     eps,
-    NB: tl.constexpr,
-    T: tl.constexpr,
-    D: tl.constexpr,
-    MBS: tl.constexpr,
+    NB,
+    T,
+    D,
+    MBS,
     BT: tl.constexpr,
     BD: tl.constexpr,
 ):