diff --git a/docs/docs/integrations/llms/sagemaker.ipynb b/docs/docs/integrations/llms/sagemaker.ipynb
index 387f91f2edd..f07a8ac311b 100644
--- a/docs/docs/integrations/llms/sagemaker.ipynb
+++ b/docs/docs/integrations/llms/sagemaker.ipynb
@@ -104,8 +104,8 @@
     "\n",
     "import boto3\n",
     "from langchain.chains.question_answering import load_qa_chain\n",
-    "from langchain_community.llms import SagemakerEndpoint\n",
-    "from langchain_community.llms.sagemaker_endpoint import LLMContentHandler\n",
+    "from langchain_aws.llms import SagemakerEndpoint\n",
+    "from langchain_aws.llms.sagemaker_endpoint import LLMContentHandler\n",
     "from langchain_core.prompts import PromptTemplate\n",
     "\n",
     "query = \"\"\"How long was Elizabeth hospitalized?\n",
@@ -174,8 +174,8 @@
     "from typing import Dict\n",
     "\n",
     "from langchain.chains.question_answering import load_qa_chain\n",
-    "from langchain_community.llms import SagemakerEndpoint\n",
-    "from langchain_community.llms.sagemaker_endpoint import LLMContentHandler\n",
+    "from langchain_aws.llms import SagemakerEndpoint\n",
+    "from langchain_aws.llms.sagemaker_endpoint import LLMContentHandler\n",
     "from langchain_core.prompts import PromptTemplate\n",
     "\n",
     "query = \"\"\"How long was Elizabeth hospitalized?\n",