diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index ae1b60752..f3005759c 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -279,14 +279,14 @@
         "max_tokens": 8192,
         "input_cost_per_token": 0.00001,
         "output_cost_per_token": 0.000028,
-        "litellm_provider": "vertex_ai-chat-models",
+        "litellm_provider": "vertex_ai-text-models",
         "mode": "completion"
     },
     "text-unicorn@001": {
         "max_tokens": 8192,
         "input_cost_per_token": 0.00001,
         "output_cost_per_token": 0.000028,
-        "litellm_provider": "vertex_ai-chat-models",
+        "litellm_provider": "vertex_ai-text-models",
         "mode": "completion"
     },
     "chat-bison": {