diff --git a/README.md b/README.md
index ede3c07..948e3c4 100644
--- a/README.md
+++ b/README.md
@@ -23,7 +23,7 @@ From the /scripts directory:
 
     Example: `poetry run python send_message.py 'I want to fly from San Francisco'`
 
-    NOTE: The workflow will pause on the 'confirm' step until the user sends a 'confirm' signal. Use the 'get_tool_data' query to see the current state of the workflow.
+    NOTE: The workflow will pause on the 'confirm' step until the user sends a 'confirm' signal. Use `poetry run python get_tool_data.py` query to see the current state of the workflow.
 
     You can send a 'confirm' signal using `poetry run python send_confirm.py`
 5. Get the conversation history summary by querying the workflow.
diff --git a/workflows/tool_workflow.py b/workflows/tool_workflow.py
index 19e3740..bec4908 100644
--- a/workflows/tool_workflow.py
+++ b/workflows/tool_workflow.py
@@ -210,7 +210,7 @@ class ToolWorkflow:
         return self.conversation_summary
 
     @workflow.query
-    def get_tool_data(self) -> Optional[str]:
+    def get_tool_data(self) -> Optional[dict]:
         return self.tool_data
 
     # Helper: generate text of the entire conversation so far