diff --git a/langchain-tool-retrieval.ipynb b/langchain-tool-retrieval.ipynb
index 6259464..1770b6c 100644
--- a/langchain-tool-retrieval.ipynb
+++ b/langchain-tool-retrieval.ipynb
@@ -39,7 +39,7 @@
     "\n",
     "# Set up OpenBB Personal Access Token from https://my.openbb.co/app/platform/pat\n",
     "from openbb import obb\n",
-    "from utils import map_openbb_collection_to_langchain_tools  # provides access to OpenBB Tools\n",
+    "from openbb_agents.utils import map_openbb_collection_to_langchain_tools  # provides access to OpenBB Tools\n",
     "obb.account.login(pat=\"\")"
    ]
   },
diff --git a/openbb-agent.ipynb b/openbb-agent.ipynb
index 88ed865..662a8dd 100644
--- a/openbb-agent.ipynb
+++ b/openbb-agent.ipynb
@@ -32,7 +32,7 @@
     "\n",
     "# Set up OpenBB Personal Access Token from https://my.openbb.co/app/platform/pat\n",
     "from openbb import obb\n",
-    "from utils import map_openbb_collection_to_langchain_tools  # provides access to OpenBB Tools\n",
+    "from openbb_agents.utils import map_openbb_collection_to_langchain_tools  # provides access to OpenBB Tools\n",
     "obb.account.login(pat=\"\")"
    ]
   },