diff --git a/generative_ai/prompts/test_resources/sample_configuration.json b/generative_ai/prompts/test_resources/sample_configuration.json
index baf1999630b..6b43b41f563 100644
--- a/generative_ai/prompts/test_resources/sample_configuration.json
+++ b/generative_ai/prompts/test_resources/sample_configuration.json
@@ -2,7 +2,7 @@
 "project": "$PROJECT_ID",
 "system_instruction_path": "gs://$CLOUD_BUCKET/sample_system_instruction.txt",
 "prompt_template_path": "gs://$CLOUD_BUCKET/sample_prompt_template.txt",
-"target_model": "gemini-1.5-flash-001",
+"target_model": "gemini-2.0-flash-001",
 "eval_metrics_types": ["safety"],
 "optimization_mode": "instruction",
 "input_data_path": "gs://$CLOUD_BUCKET/sample_prompts.jsonl",