diff --git a/tests/unit/test_fmha_v3_stage_c_full.py b/tests/unit/test_fmha_v3_stage_c_full.py
index 921420dd..b0bbda39 100644
--- a/tests/unit/test_fmha_v3_stage_c_full.py
+++ b/tests/unit/test_fmha_v3_stage_c_full.py
@@ -273,7 +273,7 @@ class FmhaV3StageC:
                 tTMEM_LOADrS_frg = cute.logical_divide(tTMEM_LOADrS, cute.make_layout(frg_tile))
                 for j in range(frg_cnt):
                     for k in range(cute.size(tTMEM_LOADrS_frg, mode=[0])):
-                        row_max = cute.arch.fmax(row_max, tTMEM_LOADrS_frg[k, j] * scale_log2)
+                        row_max = cute.arch.fmax(row_max, tTMEM_LOADrS_frg[k, j])
 
                 row_max_safe = row_max
                 if row_max == -cutlass.Float32.inf: row_max_safe = Float32(0.0)