diff --git a/notebooks/official/ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb b/notebooks/official/ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb
index 9979ea3f6..c610acd11 100644
--- a/notebooks/official/ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb
+++ b/notebooks/official/ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb
@@ -139,8 +139,8 @@
       },
       "outputs": [],
       "source": [
-        "! pip install --upgrade -q google-cloud-aiplatform \\\n",
-        "                            tensorflow -q"
+        "! pip install --upgrade --quiet google-cloud-aiplatform \\\n",
+        "                                tensorflow==2.11"
       ]
     },
     {