diff --git a/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json b/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json
index bd5cdd7c..100fb338 100644
--- a/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json
+++ b/integration-tests/models/__snapshots__/test_continue_final_message/test_llama_completion_single_prompt_continue.json
@@ -5,7 +5,7 @@
       "index": 0,
       "logprobs": null,
       "message": {
-        "content": "the mouse is much smaller than an elephant. The average elephant weight is around 6,500 lbs (3,",
+        "content": " the royal mouse? It is a little more slender and only weighs around 1.5 pounds for males and 1.3 pounds",
         "role": "assistant"
       }
     }
diff --git a/integration-tests/models/test_continue_final_message.py b/integration-tests/models/test_continue_final_message.py
index 24ae720f..01c86dcd 100644
--- a/integration-tests/models/test_continue_final_message.py
+++ b/integration-tests/models/test_continue_final_message.py
@@ -71,6 +71,6 @@ def test_llama_completion_single_prompt_continue(
     content = response["choices"][0]["message"]["content"]
     assert (
         content
-        == "the mouse is much smaller than an elephant. The average elephant weight is around 6,500 lbs (3,"
+        == " the royal mouse? It is a little more slender and only weighs around 1.5 pounds for males and 1.3 pounds"
     )
     assert response == response_snapshot