diff --git a/litellm/tests/test_sagemaker.py b/litellm/tests/test_sagemaker.py
index effffedf0..a6b202af0 100644
--- a/litellm/tests/test_sagemaker.py
+++ b/litellm/tests/test_sagemaker.py
@@ -301,6 +301,7 @@ async def test_completion_sagemaker_non_stream():
 
 
 @pytest.mark.asyncio
+@pytest.mark.flaky(retries=3, delay=1)
 async def test_completion_sagemaker_prompt_template_non_stream():
     mock_response = MagicMock()