diff --git a/litellm/tests/test_rules.py b/litellm/tests/test_rules.py
index 92ec20fe421..0c1b573b8e1 100644
--- a/litellm/tests/test_rules.py
+++ b/litellm/tests/test_rules.py
@@ -76,6 +76,7 @@ def test_post_call_rule():
         print("Got exception", e)
         print(type(e))
         print(vars(e))
+        assert e.message == "This violates LiteLLM Proxy Rules. Response too short"
         pass
     # print(f"MAKING ACOMPLETION CALL")
     # litellm.set_verbose = True