diff --git a/packages/databricks-vscode/package.json b/packages/databricks-vscode/package.json
index 3ef5b8a88..6ce55baa0 100644
--- a/packages/databricks-vscode/package.json
+++ b/packages/databricks-vscode/package.json
@@ -6,7 +6,7 @@
     "license": "LicenseRef-LICENSE",
     "version": "1.1.0",
     "engines": {
-        "vscode": "^1.76.0"
+        "vscode": "^1.78.0"
     },
     "categories": [
         "Data Science",
@@ -638,7 +638,8 @@
         ]
     },
     "extensionDependencies": [
-        "ms-python.python"
+        "ms-python.python",
+        "ms-toolsai.jupyter"
     ],
     "vsce": {
         "dependencies": false,
diff --git a/packages/databricks-vscode/src/test/e2e/wdio.conf.ts b/packages/databricks-vscode/src/test/e2e/wdio.conf.ts
index f25174b29..cc7d3663c 100644
--- a/packages/databricks-vscode/src/test/e2e/wdio.conf.ts
+++ b/packages/databricks-vscode/src/test/e2e/wdio.conf.ts
@@ -339,13 +339,17 @@ export const config: Options.Testrunner = {
         for (let i = 0; i < 2; i++) {
             try {
                 await new Promise((resolve, reject) => {
+                    const extensionDependencies =
+                        packageJson.extensionDependencies.flatMap((item) => [
+                            "--install-extension",
+                            item,
+                        ]);
                     execFile(
                         cli,
                         [
                             "--extensions-dir",
                             EXTENSION_DIR,
-                            "--install-extension",
-                            "ms-python.python",
+                            ...extensionDependencies,
                             "--install-extension",
                             VSIX_PATH,
                             "--force",