1 files changed, 11 insertions, 7 deletions
diff --git a/assets/js/theme-toggle.js b/assets/js/theme-toggle.js
index 9f0fd5a..e03fce7 100644
--- a/assets/js/theme-toggle.js
+++ b/assets/js/theme-toggle.js
@@ -45,21 +45,25 @@
 
   // Setup toggle button
   function setupToggleButton() {
-    const btn = document.getElementById('theme-toggle-btn');
+    const btn = document.getElementById('theme-switch');
     if (btn) {
       btn.addEventListener('click', toggleTheme);
-      updateToggleButtonLabel();
+      updateToggleButtonUI();
 
-      // Listen for theme changes to update button label
-      window.addEventListener('theme-changed', updateToggleButtonLabel);
+      // Listen for theme changes to update button UI
+      window.addEventListener('theme-changed', updateToggleButtonUI);
     }
   }
 
-  function updateToggleButtonLabel() {
-    const btn = document.getElementById('theme-toggle-btn');
+  function updateToggleButtonUI() {
+    const btn = document.getElementById('theme-switch');
     if (btn) {
       const current = getCurrentTheme();
-      btn.textContent = current === 'dark' ? '☀️ light' : '🌙 dark';
+      if (current === 'light') {
+        btn.classList.add('light');
+      } else {
+        btn.classList.remove('light');
+      }
     }
   }