diff --git a/module-5/memory_agent.ipynb b/module-5/memory_agent.ipynb
index aaa777738..37587c2d4 100644
--- a/module-5/memory_agent.ipynb
+++ b/module-5/memory_agent.ipynb
@@ -64,6 +64,15 @@
     "os.environ[\"LANGCHAIN_PROJECT\"] = \"langchain-academy\""
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "_set_env(\"OPENAI_API_KEY\")"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},