diff --git a/tests/local_testing/test_batch_completion_return_exceptions.py b/tests/local_testing/test_batch_completion_return_exceptions.py
index 2d2ea8675ec..24540edf318 100644
--- a/tests/local_testing/test_batch_completion_return_exceptions.py
+++ b/tests/local_testing/test_batch_completion_return_exceptions.py
@@ -8,11 +8,19 @@
 
 
 def test_batch_completion_return_exceptions_true():
-    """Test batch_completion's return_exceptions."""
+    """Test batch_completion's return_exceptions.
+    
+    With an invalid API key, we expect an error to be returned rather than raised.
+    The error type may be AuthenticationError (from API) or InternalServerError
+    (from connection issues), depending on network conditions.
+    """
     res = litellm.batch_completion(
         model="gpt-3.5-turbo",
         messages=[msg1, msg2],
         api_key="sk_xxx",  # deliberately set invalid key
     )
 
-    assert isinstance(res[0], litellm.exceptions.AuthenticationError)
+    # batch_completion should return exceptions rather than raise them
+    # Accept either AuthenticationError (API rejected key) or InternalServerError (network issues)
+    assert isinstance(res[0], (litellm.exceptions.AuthenticationError, litellm.exceptions.InternalServerError)), \
+        f"Expected AuthenticationError or InternalServerError, got {type(res[0])}"