diff --git a/docs/extras/integrations/memory/motorhead_memory_managed.ipynb b/docs/extras/integrations/memory/motorhead_memory_managed.ipynb
index e3e3e3d3cb6..bfaee89cbcc 100644
--- a/docs/extras/integrations/memory/motorhead_memory_managed.ipynb
+++ b/docs/extras/integrations/memory/motorhead_memory_managed.ipynb
@@ -21,7 +21,7 @@
    "outputs": [],
    "source": [
     "from langchain.memory.motorhead_memory import MotorheadMemory\n",
-    "from langchain.llms import OpenAI, LLMChain, PromptTemplate\n",
+    "from langchain.llms import OpenAI\nfrom langchain.chains import LLMChain\nfrom langchain.prompts import PromptTemplate\n",
     "\n",
     "template = \"\"\"You are a chatbot having a conversation with a human.\n",
     "\n",