diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 47c55ca4f3..40bc31eaaf 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -2538,6 +2538,7 @@ def test_replicate_custom_prompt_dict():
                     "content": "what is yc write 1 paragraph",
                 }
             ],
+            mock_response="hello world",
             repetition_penalty=0.1,
             num_retries=3,
         )