diff --git a/‎packages/vite/src/node/environment.ts‎
Lines changed: 3 additions & 5 deletions b/‎packages/vite/src/node/environment.ts‎
Lines changed: 3 additions & 5 deletions
@@ -23,11 +23,9 @@ export function perEnvironmentState<State>(
   const stateMap = new WeakMap<Environment, State>()
   return function (context: PluginContext) {
     const { environment } = context
-    let state = stateMap.get(environment)
-    if (!state) {
-      state = initial(environment)
-      stateMap.set(environment, state)
+    if (!stateMap.has(environment)) {
+      stateMap.set(environment, initial(environment))
     }
-    return state
+    return stateMap.get(environment)!
   }
 }
Original file line number	Diff line number	Diff line change
`@@ -23,11 +23,9 @@ export function perEnvironmentState<State>(`
`23`	`23`	`const stateMap = new WeakMap<Environment, State>()`
`24`	`24`	`return function (context: PluginContext) {`
`25`	`25`	`const { environment } = context`
`26`		`- let state = stateMap.get(environment)`
`27`		`- if (!state) {`
`28`		`- state = initial(environment)`
`29`		`- stateMap.set(environment, state)`
	`26`	`+ if (!stateMap.has(environment)) {`
	`27`	`+ stateMap.set(environment, initial(environment))`
`30`	`28`	`}`
`31`		`- return state`
	`29`	`+ return stateMap.get(environment)!`
`32`	`30`	`}`
`33`	`31`	`}`