diff --git a/src/test/java/org/opensearch/integTest/PromptHandler.java b/src/test/java/org/opensearch/integTest/PromptHandler.java
index a3f9314d..0ef03501 100644
--- a/src/test/java/org/opensearch/integTest/PromptHandler.java
+++ b/src/test/java/org/opensearch/integTest/PromptHandler.java
@@ -23,7 +23,7 @@ LLMThought llmThought() {
     }
 
     String response(String prompt) {
-        if (prompt.contains("TOOL RESPONSE: ")) {
+        if (prompt.contains("Human: TOOL RESPONSE ")) {
             return "```json{\n"
                 + "    \"thought\": \"Thought: Now I know the final answer\",\n"
                 + "    \"final_answer\": \"final answer\"\n"