diff --git a/invokeai/frontend/web/src/app/store.ts b/invokeai/frontend/web/src/app/store.ts
index d43f673bee..42306d0a82 100644
--- a/invokeai/frontend/web/src/app/store.ts
+++ b/invokeai/frontend/web/src/app/store.ts
@@ -99,7 +99,7 @@ const rootPersistConfig = getPersistConfig({
 const persistedReducer = persistReducer(rootPersistConfig, rootReducer);
 
 function buildMiddleware() {
-  if (import.meta.env.MODE === 'nodes') {
+  if (import.meta.env.MODE === 'nodes' || import.meta.env.MODE === 'package') {
     return [nodesSocketioMiddleware(), invokeMiddleware];
   } else {
     return [socketioMiddleware()];