diff --git a/.changeset/config.json b/.changeset/config.json
index 1894ce2d56..f9b3c3cf71 100644
--- a/.changeset/config.json
+++ b/.changeset/config.json
@@ -8,6 +8,7 @@
     "@inlang/cli-example",
     "@inlang/fink",
     "@inlang/paraglide-js",
+    "@inlang/paraglide-js-benchmark",
     "@inlang/paraglide-js-example-cli",
     "@inlang/paraglide-js-example-vite",
     "@inlang/paraglide-js-example-react",
diff --git a/inlang/packages/paraglide/paraglide-js/src/compiler/server/middleware.js b/inlang/packages/paraglide/paraglide-js/src/compiler/server/middleware.js
index 2f488cc0bf..42fc055760 100644
--- a/inlang/packages/paraglide/paraglide-js/src/compiler/server/middleware.js
+++ b/inlang/packages/paraglide/paraglide-js/src/compiler/server/middleware.js
@@ -168,13 +168,12 @@ function createMockAsyncLocalStorage() {
 		getStore() {
 			return currentStore;
 		},
-		run(store, callback) {
-			const previousStore = currentStore;
+		async run(store, callback) {
 			currentStore = store;
 			try {
-				return callback();
+				return await callback();
 			} finally {
-				currentStore = previousStore;
+				currentStore = undefined;
 			}
 		},
 	};