diff --git a/esbuild.config.mjs b/esbuild.config.mjs
index b3e3a68..b13282b 100644
--- a/esbuild.config.mjs
+++ b/esbuild.config.mjs
@@ -11,7 +11,7 @@ if you want to view the source, please visit the github repository of this plugi
 
 const prod = (process.argv[2] === "production");
 
-esbuild.build({
+const context = await esbuild.context({
 	banner: {
 		js: banner,
 	},
@@ -33,10 +33,16 @@ esbuild.build({
 		"@lezer/lr",
 		...builtins],
 	format: "cjs",
-	watch: !prod,
 	target: "es2018",
 	logLevel: "info",
 	sourcemap: prod ? false : "inline",
 	treeShaking: true,
 	outfile: "main.js",
-}).catch(() => process.exit(1));
+});
+
+if (prod) {
+	await context.rebuild();
+	process.exit(0);
+} else {
+	await context.watch();
+}
\ No newline at end of file