diff --git a/src/envConfig.ts b/src/envConfig.ts
index bb48f7c..73ed9f3 100644
--- a/src/envConfig.ts
+++ b/src/envConfig.ts
@@ -1,6 +1,6 @@
 import fs from 'fs';
 import path from 'path';
-import type { Plugin } from 'vite';
+import type { Plugin, ResolvedConfig } from 'vite';
 
 export function createEnvConfigContent(variables: string[], template: boolean): string {
   let templateContent = '';
@@ -26,12 +26,12 @@ export interface EnvConfigOptions {
 }
 
 export function envConfig(userOptions: Partial<EnvConfigOptions> = {}): Plugin {
-  let root: string;
+  let config: ResolvedConfig;
   return {
     name: 'vite-plugin-env-config',
 
-    configResolved(config) {
-      root = config.root;
+    configResolved(resolvedConfig) {
+      config = resolvedConfig;
     },
 
     configureServer(server) {
@@ -55,7 +55,7 @@ export function envConfig(userOptions: Partial<EnvConfigOptions> = {}): Plugin {
     closeBundle() {
       const templateContent = createEnvConfigContent(userOptions.variables || [], true);
 
-      const TEMPLATE_PATH = path.join(root, 'dist', 'env-config.template.js');
+      const TEMPLATE_PATH = path.join(config.root, config.build.outDir, 'env-config.template.js');
       fs.mkdirSync(path.dirname(TEMPLATE_PATH), { recursive: true });
       fs.writeFileSync(TEMPLATE_PATH, templateContent, 'utf8');
     },