diff --git a/v3-examples/ml-ops-examples/v3-mlflow-train-inference-e2e-example.ipynb b/v3-examples/ml-ops-examples/v3-mlflow-train-inference-e2e-example.ipynb
index 80435c9325..4717d5f65b 100644
--- a/v3-examples/ml-ops-examples/v3-mlflow-train-inference-e2e-example.ipynb
+++ b/v3-examples/ml-ops-examples/v3-mlflow-train-inference-e2e-example.ipynb
@@ -14,8 +14,9 @@
    "metadata": {},
    "source": [
     "### Prerequisites\n",
-    "- SageMaker MLflow App created (tracking server ARN required)\n",
-    "- IAM permissions for MLflow tracking and model registry\n",
+    "- SageMaker MLFlow App created (tracking server ARN required). \n",
+    "    - Follow steps here to create a serverless MLflow app managed by SageMaker: https://docs.aws.amazon.com/sagemaker/latest/dg/mlflow-app-setup.html\n",
+    "- IAM permissions for MLFlow tracking and model registry\n",
     "- AWS credentials configured"
    ]
   },
@@ -25,7 +26,7 @@
    "source": [
     "## Step 0: Install Dependencies\n",
     "\n",
-    "**Note:** There are known issues with MLflow model path resolution. Install the latest published SDK from GitHub for the latest fixes."
+    "**Note:** Version 3.4.0+ of the SageMaker SDK is required.  "
    ]
   },
   {
@@ -34,24 +35,22 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Install fix for MLflow path resolution issues\n",
-    "%pip install mlflow==3.4.0"
+    "# Install SageMaker, MLFlow & the SageMaker MLFlow Plugin \n",
+    "%pip install \"sagemaker>=3.4.0\" mlflow==3.4.0 sagemaker-mlflow==0.2.0"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### NOTE: You must restart your kernel"
+    "##### NOTE: restart your kernel"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Step 1: Configuration\n",
-    "\n",
-    "Set up MLflow tracking server and training configuration."
+    "## Step 1: Configuration\n"
    ]
   },
   {
@@ -557,7 +556,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "py3.10.14",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -571,7 +570,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.11.14"
   }
  },
  "nbformat": 4,