fix: initialize script on the case of disabling color toggle

razonyang · Jan 27, 2023 · 91616ae · 91616ae
1 parent dbe5b0a
commit 91616ae
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/assets/js/mode/index.ts b/assets/js/mode/index.ts
@@ -73,11 +73,9 @@ class ModeToggle implements Component {
       }
     });
 
-    const icon = document
-      .querySelector('.mode-item[data-color-mode="' + mode + '"] .mode-icon')
-      .cloneNode(true) as HTMLElement;
-    icon.setAttribute('id', 'modeIcon');
-    document.querySelector('#modeIcon').replaceWith(icon);
+    const icon = document.querySelector('.mode-item[data-color-mode="' + mode + '"] .mode-icon')?.cloneNode(true) as HTMLElement;
+    icon?.setAttribute('id', 'modeIcon');
+    document.querySelector('#modeIcon')?.replaceWith(icon);
   }
 
   setMode(value: string) {