diff --git a/v3-examples/inference-examples/train-inference-e2e-example.ipynb b/v3-examples/inference-examples/train-inference-e2e-example.ipynb
index 1f5cb9fdc5..683bbada79 100644
--- a/v3-examples/inference-examples/train-inference-e2e-example.ipynb
+++ b/v3-examples/inference-examples/train-inference-e2e-example.ipynb
@@ -352,7 +352,9 @@
    "outputs": [],
    "source": [
     "# Clean up resources\n",
-    "core_endpoint_config = EndpointConfig.get(endpoint_config_name=core_endpoint.endpoint_name)\n",
+    "from sagemaker.core.resources import Endpoint\n",
+    "_endpoint = Endpoint.get(endpoint_name=core_endpoint.endpoint_name)\n",
+    "core_endpoint_config = EndpointConfig.get(endpoint_config_name=_endpoint.endpoint_config_name)\n",
     "\n",
     "# Delete in the correct order\n",
     "core_model.delete()\n",