diff --git a/assets/js/zzo.js b/assets/js/zzo.js
index d822565..cb238f5 100644
--- a/assets/js/zzo.js
+++ b/assets/js/zzo.js
@@ -2,7 +2,6 @@ $(document).ready(function() {
   // theme change
   var localTheme = localStorage.getItem('theme');
   if (localTheme) {
-    $('#root').attr('class', `theme__${localTheme}`);
     $('.select-theme__item').each(function () {
       $(this).removeClass('is-active');
     });
diff --git a/layouts/_default/baseof.html b/layouts/_default/baseof.html
index 106b5c0..13db639 100644
--- a/layouts/_default/baseof.html
+++ b/layouts/_default/baseof.html
@@ -13,6 +13,12 @@
 </head>
 
 <body id="root" class="theme__{{ index .Site.Params.themeOptions 0 }}">
+    <script>
+        var localTheme = localStorage.getItem('theme');
+        if (localTheme) {
+            document.getElementById('root').className = `theme__${localTheme}`;
+        }
+    </script>
     <div id="body">        
         <div class="container wrapper">
             {{ partial "navbar/site-nav" . }}
