diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py
index c79a838835..a1349d4fa7 100644
--- a/litellm/tests/test_router.py
+++ b/litellm/tests/test_router.py
@@ -456,8 +456,11 @@ async def test_aimg_gen_on_router():
 
         router.reset()
     except Exception as e:
-        traceback.print_exc()
-        pytest.fail(f"Error occurred: {e}")
+        if "Your task failed as a result of our safety system." in str(e):
+            pass
+        else:
+            traceback.print_exc()
+            pytest.fail(f"Error occurred: {e}")
 
 
 # asyncio.run(test_aimg_gen_on_router())
diff --git a/litellm/tests/test_text_completion.py b/litellm/tests/test_text_completion.py
index b1122aa2c1..c29c299f62 100644
--- a/litellm/tests/test_text_completion.py
+++ b/litellm/tests/test_text_completion.py
@@ -2756,7 +2756,7 @@ def test_completion_chatgpt_prompt():
 
 def test_text_completion_basic():
     try:
-        print("\n test 003 with echo and logprobs \n")
+        print("\n test 003 with logprobs \n")
         litellm.set_verbose = False
         response = text_completion(
             model="gpt-3.5-turbo-instruct",