diff --git a/tdx-enhanced.js b/tdx-enhanced.js
index dabb3ff..ba1991d 100644
--- a/tdx-enhanced.js
+++ b/tdx-enhanced.js
@@ -714,8 +714,9 @@
             textBlock.style.color = "var(--txt-1)"
         })
         document.querySelectorAll("#ToggleAIAgentAssist").forEach(AIAgentButton=>{     
-        const selected = document.querySelector('input[name="aiBehavior"]:checked');
-            if (selected.value == "aiDisable") {
+        let selected = settings('get','aiBehavior');
+            console.log("AI Behavior selected:",selected.value)
+            if (selected == "aiDisable") {
                 AIAgentButton.style.display = "none"
             }  
         })