Improved code

MetaMask · Jul 25, 2023 · d0d8cc9 · d0d8cc9
1 parent f077048
commit d0d8cc9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ui/components/ui/metafox-logo/horizontal-logo.js b/ui/components/ui/metafox-logo/horizontal-logo.js
@@ -13,11 +13,11 @@ export default function MetaFoxHorizontalLogo({
   theme: themeProps,
   className,
 }) {
-  const [theme, setTheme] = useState(themeProps);
-
-  useEffect(() => {
-    setTheme(document.documentElement.getAttribute('data-theme'));
-  }, []);
+  const [theme, setTheme] = useState(() =>
+    themeProps === undefined
+      ? document.documentElement.getAttribute('data-theme')
+      : themeProps,
+  );
 
   useEffect(() => {
     if (themeProps !== undefined) {