diff --git a/vertexai/_genai/prompt_optimizer.py b/vertexai/_genai/prompt_optimizer.py
index b917c7873c..d2bcb22be8 100644
--- a/vertexai/_genai/prompt_optimizer.py
+++ b/vertexai/_genai/prompt_optimizer.py
@@ -429,7 +429,7 @@ def optimize(
 
         _OPTIMIZER_METHOD_TO_CONTAINER_URI = {
             types.PromptOptimizerMethod.VAPO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_v1_0",
-            types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/android-apo:preview_v1_0",
+            types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_android_v1_0",
         }
         container_uri = _OPTIMIZER_METHOD_TO_CONTAINER_URI.get(method)
         if not container_uri:
@@ -809,7 +809,7 @@ async def optimize(
 
         _OPTIMIZER_METHOD_TO_CONTAINER_URI = {
             types.PromptOptimizerMethod.VAPO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_v1_0",
-            types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/android-apo:preview_v1_0",
+            types.PromptOptimizerMethod.OPTIMIZATION_TARGET_GEMINI_NANO: "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_android_v1_0",
         }
         container_uri = _OPTIMIZER_METHOD_TO_CONTAINER_URI.get(method)
         if not container_uri: