diff --git a/extensions/vscode/scripts/build.js b/extensions/vscode/scripts/build.js
index 2a2fbcfff1..54935e70c6 100644
--- a/extensions/vscode/scripts/build.js
+++ b/extensions/vscode/scripts/build.js
@@ -1,9 +1,11 @@
+// @ts-check
 const path = require('path');
 const fs = require('fs');
-const treeShake = require('@kermanx/tree-shaker').treeShake;
+const esbuild = require('esbuild');
+const { treeShake } = require('@kermanx/tree-shaker');
 const minify = process.argv.includes('--minify');
 
-require('esbuild').context({
+esbuild.context({
 	entryPoints: {
 		'dist/client': './out/nodeClientMain.js',
 		'dist/server': './node_modules/@vue/language-server/bin/vue-language-server.js',