diff --git a/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi.ipynb b/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi.ipynb
index 9d5da752636..b0bb79e1d43 100644
--- a/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi.ipynb
+++ b/docs/extras/use_cases/more/agents/autonomous_agents/baby_agi.ipynb
@@ -33,7 +33,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",