diff --git a/models/demos/wormhole/mamba/tests/test_residual_block.py b/models/demos/wormhole/mamba/tests/test_residual_block.py
index 43d7fbe8e59..03a0a0df6b2 100644
--- a/models/demos/wormhole/mamba/tests/test_residual_block.py
+++ b/models/demos/wormhole/mamba/tests/test_residual_block.py
@@ -48,14 +48,14 @@ def forward(self, x):
             ModelMode.PREFILL,
             1,
             128,
-            0.9980,
+            0.9973,
         ),
         (
             "state-spaces/mamba-2.8b",
             ModelMode.DECODE,
             32,
             1,
-            0.9998,
+            0.9996,
         ),
     ),
 )