diff --git a/README.md b/README.md
index 0271130f..7de6608a 100644
--- a/README.md
+++ b/README.md
@@ -99,7 +99,7 @@ json_schema = {
 
 task = "Generate a person's information based on the following schema:"
 agent = ToolAgent(model, tokenizer, json_schema)
-generated_data = agent(task)
+generated_data = agent.run(task)
 
 print(generated_data)
 
diff --git a/swarms/agents/tool_agent.py b/swarms/agents/tool_agent.py
index ac01c5cc..58b4cde4 100644
--- a/swarms/agents/tool_agent.py
+++ b/swarms/agents/tool_agent.py
@@ -57,12 +57,11 @@ class ToolAgent(AbstractLLM):
 
         task = "Generate a person's information based on the following schema:"
         agent = ToolAgent(model, tokenizer, json_schema)
-        generated_data = agent(task)
+        generated_data = agent.run(task)
 
         print(generated_data)
 
     """
-
     def __init__(
         self,
         name: str,