diff --git a/swarms/swarms.py b/swarms/swarms.py
index 37caff7f..478cbcd5 100644
--- a/swarms/swarms.py
+++ b/swarms/swarms.py
@@ -58,7 +58,7 @@ index = faiss.IndexFlatL2(embedding_size)
 vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
 
 # ---------- Worker Node ----------
-@tool("search", return_direct=True)
+@tool
 class WorkerNode:
     def __init__(self, llm, tools, vectorstore):
         self.llm = llm