diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py
index c9e98d62d5..0d0524e5a0 100644
--- a/litellm/tests/test_alangfuse.py
+++ b/litellm/tests/test_alangfuse.py
@@ -341,6 +341,7 @@ async def test_langfuse_masked_input_output(langfuse_client):
 
 
 @pytest.mark.asyncio
+@pytest.mark.flaky(retries=3, delay=1)
 async def test_aaalangfuse_logging_metadata(langfuse_client):
     """
     Test that creates multiple traces, with a varying number of generations and sets various metadata fields
diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py
index 34c161ba4b..9b859b451a 100644
--- a/litellm/tests/test_caching.py
+++ b/litellm/tests/test_caching.py
@@ -541,6 +541,7 @@ async def test_embedding_caching_azure_individual_items_reordered():
 
 
 @pytest.mark.asyncio
+@pytest.mark.flaky(retries=3, delay=1)
 async def test_embedding_caching_base_64():
     """ """
     litellm.set_verbose = True
diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py
index b260f26ecb..25088da80f 100644
--- a/litellm/tests/test_image_generation.py
+++ b/litellm/tests/test_image_generation.py
@@ -82,6 +82,7 @@ async def test_image_generation_azure(sync_mode):
 # test_image_generation_azure()
 
 
+@pytest.mark.flaky(retries=3, delay=1)
 def test_image_generation_azure_dall_e_3():
     try:
         litellm.set_verbose = True