diff --git a/bun/uts-forester.js b/bun/uts-forester.js
index 608134f..f3d6226 100644
--- a/bun/uts-forester.js
+++ b/bun/uts-forester.js
@@ -22,8 +22,6 @@ function getAppliedMode(userPreference) {
 
 function setAppliedMode(mode) {
     document.documentElement.dataset.appliedMode = mode
-    const ninja = document.querySelector('ninja-keys')
-    ninja.setAttribute('class', mode)
 }
 
 function rotatePreferences(userPreference) {
@@ -47,6 +45,8 @@ function toggleTheme() {
 
 function search() {
     const ninja = document.querySelector('ninja-keys')
+    const mode = document.documentElement.dataset.appliedMode
+    ninja.setAttribute('class', mode)
     ninja.open()
 }