diff --git a/google/genai/tests/shared/models/test_embed.py b/google/genai/tests/shared/models/test_embed.py
index 3e1a7e018..8a6eba8b5 100644
--- a/google/genai/tests/shared/models/test_embed.py
+++ b/google/genai/tests/shared/models/test_embed.py
@@ -24,7 +24,7 @@
     pytest_helper.TestTableItem(
         name='test_embed',
         parameters=genai_types.EmbedContentParameters(
-            model='text-embedding-004',
+            model='gemini-embedding-001',
             contents='Hello world!',
         ),
     ),