diff --git a/packages/vertexai/src/requests/stream-reader.test.ts b/packages/vertexai/src/requests/stream-reader.test.ts
index d390666f4b0..eea72f9c7a6 100644
--- a/packages/vertexai/src/requests/stream-reader.test.ts
+++ b/packages/vertexai/src/requests/stream-reader.test.ts
@@ -210,7 +210,7 @@ describe('processStream', () => {
     expect(aggregatedResponse.text()).to.include('Quantum mechanics is');
     expect(
       aggregatedResponse.candidates?.[0].citationMetadata?.citations.length
-    ).to.equal(2);
+    ).to.equal(3);
     let foundCitationMetadata = false;
     for await (const response of result.stream) {
       expect(response.text()).to.not.be.empty;
diff --git a/scripts/update_vertexai_responses.sh b/scripts/update_vertexai_responses.sh
index 568612e340d..27c32a09ead 100755
--- a/scripts/update_vertexai_responses.sh
+++ b/scripts/update_vertexai_responses.sh
@@ -17,7 +17,7 @@
 # This script replaces mock response files for Vertex AI unit tests with a fresh
 # clone of the shared repository of Vertex AI test data.
 
-RESPONSES_VERSION='v2.*' # The major version of mock responses to use
+RESPONSES_VERSION='v3.*' # The major version of mock responses to use
 REPO_NAME="vertexai-sdk-test-data"
 REPO_LINK="https://github.com/FirebaseExtended/$REPO_NAME.git"