diff --git a/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi_with_agent.ipynb b/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi_with_agent.ipynb
index 910456b553c..efef081cb44 100644
--- a/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi_with_agent.ipynb
+++ b/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi_with_agent.ipynb
@@ -29,7 +29,7 @@
     "from collections import deque\n",
     "from typing import Dict, List, Optional, Any\n",
     "\n",
-    "from langchain.chains import LLMChain, OpenAI, PromptTemplate\n",
+    "from langchain.chains import LLMChain\nfrom langchain.llms import OpenAI\nfrom langchain.prompts import PromptTemplate\n",
     "from langchain.embeddings import OpenAIEmbeddings\n",
     "from langchain.llms import BaseLLM\n",
     "from langchain.vectorstores.base import VectorStore\n",