diff --git a/javascript/tagAutocomplete.js b/javascript/tagAutocomplete.js
index 45c4021..2fdd4d4 100644
--- a/javascript/tagAutocomplete.js
+++ b/javascript/tagAutocomplete.js
@@ -1059,7 +1059,10 @@ async function setup() {
     let textAreas = getTextAreas();
 
     // Add event listener to apply settings button so we can mirror the changes to our internal config
-    let applySettingsButton = gradioApp().querySelector("#tab_settings > div > .gr-button-primary");
+    let applySettingsButton = (
+        gradioApp().querySelector("#tab_settings > div > .gr-button-primary") ||
+        gradioApp().querySelector("#tab_settings > div > div > #settings_submit")
+    );
     applySettingsButton.addEventListener("click", () => {
         // Wait 500ms to make sure the settings have been applied to the webui opts object
         setTimeout(async () => { 
@@ -1158,4 +1161,4 @@ onUiUpdate(async () => {
     await syncOptions();
     // Rest of setup
     setup();
-});
\ No newline at end of file
+});