diff --git a/tests/pass_through_tests/test_gemini_with_spend.test.js b/tests/pass_through_tests/test_gemini_with_spend.test.js
index d02237fe39..4444859332 100644
--- a/tests/pass_through_tests/test_gemini_with_spend.test.js
+++ b/tests/pass_through_tests/test_gemini_with_spend.test.js
@@ -29,7 +29,7 @@ describe('Gemini AI Tests', () => {
         };
 
         const model = genAI.getGenerativeModel({
-            model: 'gemini-pro'
+            model: 'gemini-1.5-pro'
         }, requestOptions);
 
         const prompt = 'Say "hello test" and nothing else';
@@ -76,7 +76,7 @@ describe('Gemini AI Tests', () => {
         };
 
         const model = genAI.getGenerativeModel({
-            model: 'gemini-pro'
+            model: 'gemini-1.5-pro'
         }, requestOptions);
 
         const prompt = 'Say "hello test" and nothing else';
diff --git a/tests/pass_through_tests/test_local_gemini.js b/tests/pass_through_tests/test_local_gemini.js
index 7043a5ab44..1f3f7f8a0d 100644
--- a/tests/pass_through_tests/test_local_gemini.js
+++ b/tests/pass_through_tests/test_local_gemini.js
@@ -1,13 +1,13 @@
 const { GoogleGenerativeAI, ModelParams, RequestOptions } = require("@google/generative-ai");
 
 const modelParams = {
-    model: 'gemini-pro',
+    model: 'gemini-1.5-pro',
 };
   
 const requestOptions = {
     baseUrl: 'http://127.0.0.1:4000/gemini',
     customHeaders: {
-        "tags": "gemini-js-sdk,gemini-pro"
+        "tags": "gemini-js-sdk,gemini-1.5-pro"
     }
 };