diff --git a/workflows/tool_workflow.py b/workflows/tool_workflow.py
index 4a4aa7b..97893e3 100644
--- a/workflows/tool_workflow.py
+++ b/workflows/tool_workflow.py
@@ -49,7 +49,7 @@ class ToolWorkflow:
         """Execute a tool after confirmation and handle its result."""
         workflow.logger.info(f"Confirmed. Proceeding with tool: {current_tool}")
 
-        task_queue = TEMPORAL_LEGACY_TASK_QUEUE if current_tool == "BookTrain" else None
+        task_queue = TEMPORAL_LEGACY_TASK_QUEUE if current_tool in ["SearchTrain", "BookTrain"] else None
 
         try:
             dynamic_result = await workflow.execute_activity(