diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py
index cc220e46e..db84234c7 100644
--- a/tests/local_testing/test_streaming.py
+++ b/tests/local_testing/test_streaming.py
@@ -3333,7 +3333,7 @@ async def test_acompletion_function_call_with_streaming(model):
                 validate_final_streaming_function_calling_chunk(chunk=chunk)
             idx += 1
         # raise Exception("it worked! ")
-    except litellm.InternalServerError:
+    except litellm.InternalServerError or litellm.ServiceUnavailableError:
         pass
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")