diff --git a/javascript/tagAutocomplete.js b/javascript/tagAutocomplete.js
index 698f6c9..5240eea 100644
--- a/javascript/tagAutocomplete.js
+++ b/javascript/tagAutocomplete.js
@@ -383,7 +383,7 @@ function addResultsToList(textArea, results, tagword, resetList) {
     // Find right colors from config
     let tagFileName = CFG.tagFile.split(".")[0];
     let tagColors = CFG.colorMap;
-    let mode = gradioApp().querySelector('.dark') ? 0 : 1;
+    let mode = document.querySelector('.dark') ? 0 : 1;
     let nextLength = Math.min(results.length, resultCount + CFG.resultStepLength);
 
     for (let i = resultCount; i < nextLength; i++) {