diff --git a/cypress/integration/plugins/dashboards-assistant/chatbot_interaction_trace_spec.js b/cypress/integration/plugins/dashboards-assistant/chatbot_interaction_trace_spec.js
index 3b11bb8da..6d8a445f1 100644
--- a/cypress/integration/plugins/dashboards-assistant/chatbot_interaction_trace_spec.js
+++ b/cypress/integration/plugins/dashboards-assistant/chatbot_interaction_trace_spec.js
@@ -74,7 +74,7 @@ if (Cypress.env('DASHBOARDS_ASSISTANT_ENABLED')) {
         cy.get('@tracePage')
           .find('.euiAccordion')
           // tool name
-          .contains('Step 1 - CatIndexTool')
+          .contains('Step 2 - CatIndexTool')
           .click({ force: true });
 
         // tool output
diff --git a/cypress/support/assistant-dummy-llm.js b/cypress/support/assistant-dummy-llm.js
index f80ef3803..b317d8192 100644
--- a/cypress/support/assistant-dummy-llm.js
+++ b/cypress/support/assistant-dummy-llm.js
@@ -10,7 +10,7 @@ const suggestionJson = require('../fixtures/plugins/dashboards-assistant/suggest
 const MATCH_AGENT_FRAMEWORK_PROMPT =
   'Assistant is designed to be able to assist with a wide range of tasks';
 const MATCH_SUGGESTION_PROMPT = 'You are an AI that only speaks JSON';
-const TOOL_RESPONSE = 'TOOL RESPONSE of ';
+const TOOL_RESPONSE = 'TOOL RESPONSE of CatIndexTool:';
 
 const server = http.createServer((req, res) => {
   // Set the content type to JSON