diff --git a/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py b/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py
index 262c55efc5d..cfdea73cefa 100644
--- a/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py
+++ b/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py
@@ -235,6 +235,7 @@ class TestManagedFilesAPI(ManagedFilesBase, UserKeyTestMixin):
     # Tests
     # ------------------------------------------------------------------
 
+    @pytest.mark.flaky(reruns=5)
     @pytest.mark.parametrize(
         "model_name",
         get_batch_model_names(),