diff --git a/src/renderer/env.d.ts b/src/renderer/env.d.ts
index e09f6db1..4059d462 100644
--- a/src/renderer/env.d.ts
+++ b/src/renderer/env.d.ts
@@ -1,3 +1,4 @@
+/// <reference types="vite/client" />
 import type { DesktopApi } from "@shared/contracts";
 
 declare global {
diff --git a/src/renderer/state/store.ts b/src/renderer/state/store.ts
index be6e8bd1..91e7aea5 100644
--- a/src/renderer/state/store.ts
+++ b/src/renderer/state/store.ts
@@ -39,7 +39,7 @@ export const useAppStore = create<AppState>()(
     })),
     {
       name: "AgentCompany",
-      enabled: process.env.NODE_ENV === "development",
+      enabled: import.meta.env.DEV,
     }
   )
 );