diff --git a/test/registered/layers/mamba/test_mamba_ssm.py b/test/registered/layers/mamba/test_mamba_ssm.py
index e2532e9c6..0178bfa2c 100644
--- a/test/registered/layers/mamba/test_mamba_ssm.py
+++ b/test/registered/layers/mamba/test_mamba_ssm.py
@@ -99,7 +99,7 @@ def test_selective_state_update(dim, dstate, has_z, itype):
 
     rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (5e-3, 1e-2)
     if itype == torch.bfloat16:
-        rtol, atol = 1e-2, 5e-2
+        rtol, atol = 1e-2, 1e-1
         if torch.version.hip:
             atol *= 2
     # set seed