diff --git a/src/build.ts b/src/build.ts
index 0a5b7c0..550e5e9 100644
--- a/src/build.ts
+++ b/src/build.ts
@@ -5,6 +5,7 @@ import { InlineConfig, resolveConfig } from './config'
  * Bundles the electron app for production.
  */
 export async function build(inlineConfig: InlineConfig = {}): Promise<void> {
+  process.env.NODE_ENV_ELECTRON_VITE = 'production'
   const config = await resolveConfig(inlineConfig, 'build', 'production')
   if (config.config) {
     const mainViteConfig = config.config?.main
diff --git a/src/server.ts b/src/server.ts
index 9e324f3..85d4ac0 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -13,6 +13,7 @@ import { resolveHostname } from './utils'
 import { startElectron } from './electron'
 
 export async function createServer(inlineConfig: InlineConfig = {}): Promise<void> {
+  process.env.NODE_ENV_ELECTRON_VITE = 'development'
   const config = await resolveConfig(inlineConfig, 'serve', 'development')
   if (config.config) {
     const logger = createLogger(inlineConfig.logLevel)