diff --git a/litellm/tests/test_rules.py b/litellm/tests/test_rules.py
index f359a6de353..7e2d7b81962 100644
--- a/litellm/tests/test_rules.py
+++ b/litellm/tests/test_rules.py
@@ -127,8 +127,8 @@ def test_post_call_rule_streaming():
         print(type(e))
         print(vars(e))
         assert (
-            e.message
-            == "OpenAIException - This violates LiteLLM Proxy Rules. Response too short"
+            "OpenAIException - This violates LiteLLM Proxy Rules. Response too short"
+            in e.message
         )