diff --git a/SampleCode/Python/sample_rag_langchain.ipynb b/SampleCode/Python/sample_rag_langchain.ipynb
index 5457048..16ec0ed 100644
--- a/SampleCode/Python/sample_rag_langchain.ipynb
+++ b/SampleCode/Python/sample_rag_langchain.ipynb
@@ -120,6 +120,8 @@
    "source": [
     "# Embed the splitted documents and insert into Azure Search vector store\n",
     "\n",
+    "import openai \n",
+    "openai.api_base = os.getenv(\"AZURE_OPENAI_ENDPOINT\")\n",
     "aoai_embeddings = AzureOpenAIEmbeddings(\n",
     "    azure_deployment=\"<Azure OpenAI embeddings model>\",\n",
     "    openai_api_version=\"<Azure OpenAI API version>\",  # e.g., \"2023-07-01-preview\"\n",