diff --git a/integration-tests/models/test_flash_pali_gemma.py b/integration-tests/models/test_flash_pali_gemma.py
index 94b7b89d..e1fad0d3 100644
--- a/integration-tests/models/test_flash_pali_gemma.py
+++ b/integration-tests/models/test_flash_pali_gemma.py
@@ -34,6 +34,4 @@ async def test_flash_pali_gemma(flash_pali_gemma, response_snapshot):
     inputs = f"Where is the cow standing?\n![]({cow})"
     response = await flash_pali_gemma.generate(inputs, max_new_tokens=20)
 
-    # TODO: update this! this is incorrect and just to show the current state of the test
-    assert response.generated_text == " - HDS"
-    # assert response.generated_text == "\nbeach"
+    assert response.generated_text == "\nbeach"