diff --git a/.github/actions/spelling/allow.txt b/.github/actions/spelling/allow.txt
index 084936e4cc3..6ec7e7edcc9 100644
--- a/.github/actions/spelling/allow.txt
+++ b/.github/actions/spelling/allow.txt
@@ -359,6 +359,7 @@ cer
 cfbundle
 chatbots
 chromadb
+cimg
 claude
 clickable
 cmap
@@ -378,6 +379,7 @@ constexpr
 corpuses
 csa
 cse
+ctd
 cupertino
 dask
 dataframe
diff --git a/gemini/prompts/prompt_optimizer/vapo_lib.py b/gemini/prompts/prompt_optimizer/vapo_lib.py
index 2365f801ccd..b60af6b764c 100644
--- a/gemini/prompts/prompt_optimizer/vapo_lib.py
+++ b/gemini/prompts/prompt_optimizer/vapo_lib.py
@@ -21,15 +21,12 @@
 from IPython.core.display import DisplayHandle
 from IPython.display import HTML, display
 from google.cloud import aiplatform, storage
-from google.colab import output
 import ipywidgets as widgets
 import jinja2
 import jinja2.meta
 import pandas as pd
 from tensorflow.io import gfile
 
-output.enable_custom_widget_manager()
-
 
 def is_target_required_metric(eval_metric: str) -> bool:
     """Check if the metric requires the target label."""
diff --git a/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb b/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb
index a17a30b1c19..d56b179d73d 100644
--- a/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb
+++ b/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb
@@ -291,8 +291,6 @@
         "import json\n",
         "import time\n",
         "\n",
-        "from google.colab import auth\n",
-        "\n",
         "timestamp = datetime.datetime.now().strftime(\"%Y-%m-%dT%H:%M:%S\")\n",
         "display_name = f\"pt_{timestamp}\"\n",
         "\n",