diff --git a/examples/notebooks/beam-ml/anomaly_detection/anomaly_detection_timesfm.ipynb b/examples/notebooks/beam-ml/anomaly_detection/anomaly_detection_timesfm.ipynb
index 034dca22a42b..e232daf02d3e 100644
--- a/examples/notebooks/beam-ml/anomaly_detection/anomaly_detection_timesfm.ipynb
+++ b/examples/notebooks/beam-ml/anomaly_detection/anomaly_detection_timesfm.ipynb
@@ -1785,7 +1785,7 @@
         "# =================================================================\n",
         "# Classify with LLM and Create Clean Data for Finetuning\n",
         "# =================================================================\n",
-        "api_key = \"AIzaSyCB_g6tq3eBFtB3BsshdGotLkUkTsCyApY\" #userdata.get('GEMINI_API_KEY')\n",
+        "api_key = \"userdata.get('GEMINI_API_KEY') # @param {type:'string'} \n",
         "\n",
         "llm_classifier = (\n",
         "    \"LLMClassifierAndImputer\" >> beam.ParDo(\n",
@@ -2709,4 +2709,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}