From 8a99d34af94e5f40fdfa67aaf4bb0d8f7b86df5a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Sat, 13 Sep 2025 21:33:16 -0500
Subject: [PATCH] fix: check intialization this mount

---
 packages/varuna/src/index.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/varuna/src/index.ts b/packages/varuna/src/index.ts
index 1f95eb763..a7108c2fa 100644
--- a/packages/varuna/src/index.ts
+++ b/packages/varuna/src/index.ts
@@ -1,6 +1,6 @@
 import { produce } from "immer";
 import { merge } from "lodash-es";
-import { useEffect, useState } from "react";
+import { useEffect, useRef, useState } from "react";
 
 export class Varuna<Type, Arguments = void> {
   private listeners = new Set<(state: Type) => void>();
@@ -39,13 +39,16 @@ export class Varuna<Type, Arguments = void> {
   }
 
   useInitialization(...args: Arguments[]) {
-    if (this.initialized) return;
+    const initializedThisMount = useRef(false);
+
+    if (initializedThisMount.current) return;
 
     if (typeof this.creator !== "function") {
       throw new Error("Provider must be used with a function creator");
     }
 
     this.initialize((this.creator as (...args: Arguments[]) => Type)(...args));
+    initializedThisMount.current = true;
   }
 
   private dispatch() {