diff --git a/integration-tests/models/test_mamba.py b/integration-tests/models/test_mamba.py
index fcd2d59f..1fba2100 100644
--- a/integration-tests/models/test_mamba.py
+++ b/integration-tests/models/test_mamba.py
@@ -54,7 +54,7 @@ async def test_mamba_all_params(fused_kernel_mamba, response_snapshot):
 
 @pytest.mark.asyncio
 @pytest.mark.private
-async def test_mamba_load(fused_kernel_mamba, generate_load, response_snapshot):
+async def test_mamba_load(fused_kernel_mamba, generate_load, generous_response_snapshot):
     responses = await generate_load(
         fused_kernel_mamba, "What is Deep Learning?", max_new_tokens=10, n=4
     )