diff --git a/integration-tests/models/__snapshots__/test_mllama/test_mllama_load.json b/integration-tests/models/__snapshots__/test_mllama/test_mllama_load.json
index 65154e89..58988999 100644
--- a/integration-tests/models/__snapshots__/test_mllama/test_mllama_load.json
+++ b/integration-tests/models/__snapshots__/test_mllama/test_mllama_load.json
@@ -6,7 +6,7 @@
         "index": 0,
         "logprobs": null,
         "message": {
-          "content": "In a small town, a chicken named Cluck",
+          "content": "In a small village, a chicken named Cluck",
           "name": null,
           "role": "assistant",
           "tool_calls": null
@@ -32,7 +32,7 @@
         "index": 0,
         "logprobs": null,
         "message": {
-          "content": "In a small town, a chicken named Cluck",
+          "content": "In a small village, a chicken named Cluck",
           "name": null,
           "role": "assistant",
           "tool_calls": null
diff --git a/integration-tests/models/__snapshots__/test_mllama/test_mllama_simpl.json b/integration-tests/models/__snapshots__/test_mllama/test_mllama_simpl.json
index 14ca3f4e..1f9ed8b2 100644
--- a/integration-tests/models/__snapshots__/test_mllama/test_mllama_simpl.json
+++ b/integration-tests/models/__snapshots__/test_mllama/test_mllama_simpl.json
@@ -5,7 +5,7 @@
       "index": 0,
       "logprobs": null,
       "message": {
-        "content": "In a small town, a chicken named Cluck",
+        "content": "In a small village, a chicken named Cluck",
         "name": null,
         "role": "assistant",
         "tool_calls": null
diff --git a/integration-tests/models/test_mllama.py b/integration-tests/models/test_mllama.py
index 90b2bff1..e40f0be6 100644
--- a/integration-tests/models/test_mllama.py
+++ b/integration-tests/models/test_mllama.py
@@ -47,7 +47,8 @@ async def test_mllama_simpl(mllama, response_snapshot):
         "total_tokens": 60,
     }
     assert (
-        response.choices[0].message.content == "In a small town, a chicken named Cluck"
+        response.choices[0].message.content
+        == "In a small village, a chicken named Cluck"
     )
     assert response == response_snapshot
 
@@ -86,7 +87,7 @@ async def test_mllama_load(mllama, generate_load, response_snapshot):
     generated_texts = [response.choices[0].message.content for response in responses]
 
     # XXX: TODO: Fix this test.
-    assert generated_texts[0] == "In a small town, a chicken named Cluck"
+    assert generated_texts[0] == "In a small village, a chicken named Cluck"
     assert len(generated_texts) == 2
     assert generated_texts, all(
         [text == generated_texts[0] for text in generated_texts]