diff --git a/docs/extras/modules/memory/agent_with_memory.ipynb b/docs/extras/modules/memory/agent_with_memory.ipynb
index 11cf34dc288..1ece76969d2 100644
--- a/docs/extras/modules/memory/agent_with_memory.ipynb
+++ b/docs/extras/modules/memory/agent_with_memory.ipynb
@@ -29,7 +29,7 @@
    "source": [
     "from langchain.agents import ZeroShotAgent, Tool, AgentExecutor\n",
     "from langchain.memory import ConversationBufferMemory\n",
-    "from langchain.llms import OpenAI, LLMChain\n",
+    "from langchain.llms import OpenAI\nfrom langchain.chains import LLMChain\n",
     "from langchain.utilities import GoogleSearchAPIWrapper"
    ]
   },