diff --git a/.vscode/launch.json b/.vscode/launch.json
index 18b3b5721..b14e2ee22 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -27,7 +27,7 @@
             "request": "attach",
             "skipFiles": ["<node_internals>/**"],
             "sourceMaps": true,
-            "outFiles": ["${workspaceFolder}/packages/schema/bundle/**/*.js"]
+            "outFiles": ["${workspaceFolder}/packages/schema/dist/**/*.js"]
         }
     ]
 }
diff --git a/packages/schema/package.json b/packages/schema/package.json
index efb704ba1..3448fabdd 100644
--- a/packages/schema/package.json
+++ b/packages/schema/package.json
@@ -74,7 +74,7 @@
     "bin": {
         "zenstack": "bin/cli"
     },
-    "main": "./bundle/extension.js",
+    "main": "./dist/extension.js",
     "scripts": {
         "vscode:publish": "vsce publish --no-dependencies",
         "vscode:prerelease": "vsce publish --no-dependencies --pre-release",
diff --git a/packages/schema/src/extension.ts b/packages/schema/src/extension.ts
index a3e19d7f8..20d4b37b7 100644
--- a/packages/schema/src/extension.ts
+++ b/packages/schema/src/extension.ts
@@ -18,7 +18,7 @@ export function deactivate(): Thenable<void> | undefined {
 }
 
 function startLanguageClient(context: vscode.ExtensionContext): LanguageClient {
-    const serverModule = context.asAbsolutePath(path.join('bundle', 'language-server', 'main'));
+    const serverModule = context.asAbsolutePath(path.join('dist', 'language-server', 'main'));
     // The debug options for the server
     // --inspect=6009: runs the server in Node's Inspector mode so VS Code can attach to the server for debugging.
     // By setting `process.env.DEBUG_BREAK` to a truthy value, the language server will wait until a debugger is attached.
diff --git a/packages/schema/tsconfig.json b/packages/schema/tsconfig.json
index 0cc75a43a..fe7d16026 100644
--- a/packages/schema/tsconfig.json
+++ b/packages/schema/tsconfig.json
@@ -3,6 +3,6 @@
     "compilerOptions": {
         "outDir": "dist"
     },
-    "include": ["src/**/*.ts"],
-    "exclude": ["src/extension.ts"]
+    "include": ["src/**/*.ts"]
+    // "exclude": ["src/extension.ts"]
 }