From 2714d0a3790e1c1422649f8e4d773129498be75b Mon Sep 17 00:00:00 2001
From: Mathieu Acthernoene <zoontek@gmail.com>
Date: Fri, 6 Dec 2024 15:58:26 +0100
Subject: [PATCH] Keep else

---
 packages/lake/src/hooks/usePersistedState.ts | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/packages/lake/src/hooks/usePersistedState.ts b/packages/lake/src/hooks/usePersistedState.ts
index 27025b3c8..53aef495a 100644
--- a/packages/lake/src/hooks/usePersistedState.ts
+++ b/packages/lake/src/hooks/usePersistedState.ts
@@ -40,18 +40,16 @@ export const usePersistedState = <T>(key: string, defaultValue: T) => {
       if (typeof value !== "function") {
         const rawValue = value != null ? stringifyValue(value) : null;
         setItem(key, rawValue);
-        return setRawValue(rawValue);
+        setRawValue(rawValue);
+      } else {
+        setRawValue(prevState => {
+          const prevValue = parseRawValue(prevState, stableDefaultValue);
+          const nextValue = (value as (prevState: T) => T)(prevValue);
+          const rawValue = stringifyValue(nextValue);
+          setItem(key, rawValue);
+          return rawValue;
+        });
       }
-
-      setRawValue(prevState => {
-        const nextValue = (value as (prevState: T) => T)(
-          parseRawValue(prevState, stableDefaultValue),
-        );
-
-        const rawValue = stringifyValue(nextValue);
-        setItem(key, rawValue);
-        return rawValue;
-      });
     },
     [key, stableDefaultValue],
   );