diff --git a/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java b/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java
index 2d081403698..1ad1f09f003 100644
--- a/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java
+++ b/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java
@@ -72,7 +72,7 @@ public static BatchPredictionJob createBatchPredictionGeminiJobSample(
       String modelName =
           String.format(
               "projects/%s/locations/%s/publishers/google/models/%s",
-              project, location, "gemini-2.0-flash-001");
+              project, location, "gemini-2.5-flash");
 
       BatchPredictionJob batchPredictionJob =
           BatchPredictionJob.newBuilder()