diff --git a/docs/extras/integrations/memory/motorhead_memory.ipynb b/docs/extras/integrations/memory/motorhead_memory.ipynb
index 8402315a07a..87ded21c3e6 100644
--- a/docs/extras/integrations/memory/motorhead_memory.ipynb
+++ b/docs/extras/integrations/memory/motorhead_memory.ipynb
@@ -20,7 +20,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",