diff --git a/notebooks/lesson_3_semantic.ipynb b/notebooks/lesson_3_semantic.ipynb
index 505ccc0..c70a1a8 100644
--- a/notebooks/lesson_3_semantic.ipynb
+++ b/notebooks/lesson_3_semantic.ipynb
@@ -43,7 +43,11 @@
     "from getpass import getpass\n",
     "\n",
     "if not os.environ.get(\"ANTHROPIC_API_KEY\"):\n",
-    "    os.environ[\"ANTHROPIC_API_KEY\"] = getpass(\"ANTHROPIC_API_KEY: \")"
+    "    # For the LLM\n",
+    "    os.environ[\"ANTHROPIC_API_KEY\"] = getpass(\"ANTHROPIC_API_KEY: \")\n",
+    "if not os.environ.get(\"OPENAI_API_KEY\"):\n",
+    "    # For the embedding model\n",
+    "    os.environ[\"OPENAI_API_KEY\"] = getpass(\"OPENAI_API_KEY: \")"
    ]
   },
   {
@@ -177,7 +181,7 @@
     "from langgraph.store.memory import InMemoryStore\n",
     "\n",
     "# Memory store\n",
-    "store = InMemoryStore()"
+    "store = InMemoryStore(index={\"embed\": \"openai:text-embedding-3-small\"})"
    ]
   },
   {