diff --git a/integration-tests/models/__snapshots__/test_flash_gemma/test_flash_gemma.json b/integration-tests/models/__snapshots__/test_flash_gemma/test_flash_gemma_simple.json
similarity index 100%
rename from integration-tests/models/__snapshots__/test_flash_gemma/test_flash_gemma.json
rename to integration-tests/models/__snapshots__/test_flash_gemma/test_flash_gemma_simple.json
diff --git a/integration-tests/models/__snapshots__/test_flash_llama/test_flash_llama.json b/integration-tests/models/__snapshots__/test_flash_llama/test_flash_llama_simple.json
similarity index 100%
rename from integration-tests/models/__snapshots__/test_flash_llama/test_flash_llama.json
rename to integration-tests/models/__snapshots__/test_flash_llama/test_flash_llama_simple.json