diff --git a/litellm/tests/test_proxy_exception_mapping.py b/litellm/tests/test_proxy_exception_mapping.py
index a2c7952586..82957b658c 100644
--- a/litellm/tests/test_proxy_exception_mapping.py
+++ b/litellm/tests/test_proxy_exception_mapping.py
@@ -167,8 +167,9 @@ def test_chat_completion_exception_any_model(client):
         openai_exception = openai_client._make_status_error_from_response(
             response=response
         )
-        print("Exception raised=", openai_exception)
         assert isinstance(openai_exception, openai.BadRequestError)
+        _error_message = openai_exception.message
+        assert "Invalid model name passed in model=Lite-GPT-12" in str(_error_message)
 
     except Exception as e:
         pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
@@ -195,6 +196,8 @@ def test_embedding_exception_any_model(client):
         )
         print("Exception raised=", openai_exception)
         assert isinstance(openai_exception, openai.BadRequestError)
+        _error_message = openai_exception.message
+        assert "Invalid model name passed in model=Lite-GPT-12" in str(_error_message)
 
     except Exception as e:
         pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")