diff --git a/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb b/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb
index 994556ce76..96d09be4b1 100644
--- a/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb
+++ b/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb
@@ -39,11 +39,11 @@
    "source": [
     "from typing import Dict, List\n",
     "from langchain.embeddings import SagemakerEndpointEmbeddings\n",
-    "from langchain.llms.sagemaker_endpoint import ContentHandlerBase\n",
+    "from langchain.embeddings.sagemaker_endpoint import EmbeddingsContentHandler\n",
     "import json\n",
     "\n",
     "\n",
-    "class ContentHandler(ContentHandlerBase):\n",
+    "class ContentHandler(EmbeddingsContentHandler):\n",
     "    content_type = \"application/json\"\n",
     "    accepts = \"application/json\"\n",
     "\n",