diff --git a/README.md b/README.md
index 997bc797..139fcaf0 100644
--- a/README.md
+++ b/README.md
@@ -78,7 +78,7 @@ agent.run("Generate a 10,000 word blog on health and wellness.")
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-from swarms import ToolAgent
+from swarms.agents.tool_agent import ToolAgent
 
 
 model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-12b")