diff --git a/v3-examples/inference-examples/train-inference-e2e-example.ipynb b/v3-examples/inference-examples/train-inference-e2e-example.ipynb
index ee9b0ac0c7..1f5cb9fdc5 100644
--- a/v3-examples/inference-examples/train-inference-e2e-example.ipynb
+++ b/v3-examples/inference-examples/train-inference-e2e-example.ipynb
@@ -61,7 +61,8 @@
     "TRAINING_JOB_PREFIX = \"e2e-v3-pytorch\"\n",
     "\n",
     "# AWS Configuration\n",
-    "AWS_REGION = Session.boto_region_name\n",
+    "session = Session()\n",
+    "AWS_REGION = session.boto_region_name\n",
     "PYTORCH_TRAINING_IMAGE = f\"763104351884.dkr.ecr.{AWS_REGION}.amazonaws.com/pytorch-training:1.13.1-cpu-py39\"\n",
     "\n",
     "# Generate unique identifiers\n",