diff --git a/timm/layers/ml_decoder.py b/timm/layers/ml_decoder.py
index 5f8f403c0..32811cb85 100644
--- a/timm/layers/ml_decoder.py
+++ b/timm/layers/ml_decoder.py
@@ -349,7 +349,7 @@ def __init__(
         learnable_class_embed: bool = False,
         embed_drop: float = 0.1,
         embed_norm: bool = True,
-        k_norm: bool = False,
+        qk_norm: bool = False,
         attn_drop: float = 0.1,
         mlp_ratio: float = 8/3,
         proj_drop: float = 0.1,