diff --git a/e2e/smoke-tests/tests/modular.test.ts b/e2e/smoke-tests/tests/modular.test.ts
index c6cd0b88c8e..c8f6afe779d 100644
--- a/e2e/smoke-tests/tests/modular.test.ts
+++ b/e2e/smoke-tests/tests/modular.test.ts
@@ -86,13 +86,9 @@ import {
   StorageReference,
   deleteObject
 } from 'firebase/storage';
-import {
-  getGenerativeModel,
-  getAI,
-  AI,
-  VertexAIBackend
-} from 'firebase/vertexai';
+import { getGenerativeModel, getAI, AI, VertexAIBackend } from 'firebase/ai';
 import { getDataConnect, DataConnect } from 'firebase/data-connect';
+// @ts-ignore
 import { config, testAccount } from '../firebase-config';
 import 'jest';
 
diff --git a/e2e/smoke-tests/tsconfig.json b/e2e/smoke-tests/tsconfig.json
new file mode 100644
index 00000000000..c09f694edf7
--- /dev/null
+++ b/e2e/smoke-tests/tsconfig.json
@@ -0,0 +1,7 @@
+{
+  "extends": "../../config/tsconfig.base.json",
+  "compilerOptions": {
+    "outDir": "dist",
+    "typeRoots": ["./node_modules/@types"]
+  }
+}