diff --git a/src/server.ts b/src/server.ts
index 96df13a8e4086fedb884496f1aca1480d2fd66c3..16fa63992d08b67809b93144121e314cf0fae9b2 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -565,12 +565,11 @@       if (state.modules.jit && exports.mode === 'jit') {
         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)) {