fix up logic

pacocoursey · Aug 18, 2023 · 1579445 · 1579445
1 parent 4ee5ee6
commit 1579445
Showing 1 changed file with 9 additions and 13 deletions.
diff --git a/packages/next-themes/src/index.tsx b/packages/next-themes/src/index.tsx
@@ -82,19 +82,15 @@ const Theme: React.FC<ThemeProviderProps> = ({
 
   const setTheme = useCallback(
     theme => {
-      setThemeState(prevTheme => {
-        const currentTheme = prevTheme ?? defaultTheme
-        const newTheme = typeof theme === 'function' ? theme(currentTheme) : currentTheme
-
-        // Save to storage
-        try {
-          localStorage.setItem(storageKey, newTheme)
-        } catch (e) {
-          // Unsupported
-        }
-
-        return newTheme
-      })
+      const newTheme = typeof theme === 'function' ? theme(theme) : theme
+      setThemeState(newTheme)
+
+      // Save to storage
+      try {
+        localStorage.setItem(storageKey, newTheme)
+      } catch (e) {
+        // Unsupported
+      }
     },
     [forcedTheme]
   )