diff --git a/docs/extras/modules/memory/agent_with_memory_in_db.ipynb b/docs/extras/modules/memory/agent_with_memory_in_db.ipynb
index f8e527f45a0..adf48b9e040 100644
--- a/docs/extras/modules/memory/agent_with_memory_in_db.ipynb
+++ b/docs/extras/modules/memory/agent_with_memory_in_db.ipynb
@@ -37,7 +37,7 @@
     "from langchain.memory import ConversationBufferMemory\n",
     "from langchain.memory.chat_memory import ChatMessageHistory\n",
     "from langchain.memory.chat_message_histories import RedisChatMessageHistory\n",
-    "from langchain.llms import OpenAI, LLMChain\n",
+    "from langchain.llms import OpenAI\nfrom langchain.chains import LLMChain\n",
     "from langchain.utilities import GoogleSearchAPIWrapper"
    ]
   },