diff --git a/README.md b/README.md
index 7de6608a..2f9883a6 100644
--- a/README.md
+++ b/README.md
@@ -98,7 +98,7 @@ json_schema = {
 }
 
 task = "Generate a person's information based on the following schema:"
-agent = ToolAgent(model, tokenizer, json_schema)
+agent = ToolAgent(model=model, tokenizer=tokenizer, json_schema=json_schema)
 generated_data = agent.run(task)
 
 print(generated_data)
diff --git a/swarms/agents/tool_agent.py b/swarms/agents/tool_agent.py
index 7dade650..594a1863 100644
--- a/swarms/agents/tool_agent.py
+++ b/swarms/agents/tool_agent.py
@@ -56,11 +56,10 @@ class ToolAgent(AbstractLLM):
         }
 
         task = "Generate a person's information based on the following schema:"
-        agent = ToolAgent(model, tokenizer, json_schema)
+        agent = ToolAgent(model=model, tokenizer=tokenizer, json_schema=json_schema)
         generated_data = agent.run(task)
 
         print(generated_data)
-
     """
     def __init__(
         self,