diff --git a/tests/integration/memory/vertex_ai_memory_bank_service_test.ts b/tests/integration/memory/vertex_ai_memory_bank_service_test.ts
index 9e87fca64..637c0a75c 100644
--- a/tests/integration/memory/vertex_ai_memory_bank_service_test.ts
+++ b/tests/integration/memory/vertex_ai_memory_bank_service_test.ts
@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import {Client} from '@google-cloud/vertexai/build/src/genai/client.js';
+import {Client} from '@google-cloud/vertexai';
 import {
   createEvent,
   InMemorySessionService,
diff --git a/tests/integration/tools/agent_tool_vertexai_test.ts b/tests/integration/tools/agent_tool_vertexai_test.ts
index 2ea7e6a68..f0dbbafe0 100644
--- a/tests/integration/tools/agent_tool_vertexai_test.ts
+++ b/tests/integration/tools/agent_tool_vertexai_test.ts
@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import {Client} from '@google-cloud/vertexai/build/src/genai/client.js';
+import {Client} from '@google-cloud/vertexai';
 import {Sessions} from '@google-cloud/vertexai/build/src/genai/sessions.js';
 import {
   AgentTool,