diff --git a/src/server.ts b/src/server.ts
index 16fa639..96df13a 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -565,12 +565,13 @@ async function createProjectService(
         state.jit = true
         userVariants = exports.variants
         exports.variants = []
-        userMode = exports.mode
-        delete exports.mode
       } else {
         state.jit = false
       }
 
+      userMode = exports.mode
+      delete exports.mode
+
       // inject JIT `matchUtilities` function
       if (Array.isArray(exports.plugins)) {
         userPlugins = exports.plugins