diff --git a/tdx-enhanced.js b/tdx-enhanced.js
index ba1991d..4030985 100644
--- a/tdx-enhanced.js
+++ b/tdx-enhanced.js
@@ -686,7 +686,6 @@
             }
         }
     }
-
     function parseOtherElements() {
         let path = document.location.pathname;
 
@@ -713,14 +712,13 @@
         document.querySelectorAll("div.moreToggle *:not(a), div.lessToggle *:not(a), div.wrap-text *:not(a)").forEach(textBlock=>{
             textBlock.style.color = "var(--txt-1)"
         })
-        document.querySelectorAll("#ToggleAIAgentAssist").forEach(AIAgentButton=>{     
+        document.querySelectorAll("#ToggleAIAgentAssist, .tdx-view-ticket-summary").forEach(AIAgentButton=>{     
         let selected = settings('get','aiBehavior');
-            console.log("AI Behavior selected:",selected.value)
             if (selected == "aiDisable") {
                 AIAgentButton.style.display = "none"
             }  
         })
-
+              
         let desktopLayout = settings('get','layout')
         let desktop = document.querySelector("#divContent")
         if (desktop && desktopLayout) {
@@ -1227,6 +1225,22 @@
             if (t.classList.contains("tdx-right-side-panel")) {
                 parseOtherElements()
             }
+            if (t.classList.contains("sr-alert-region")){ // magic class
+                document.querySelectorAll("#divFieldSuggestionsResponsible, #divFieldSuggestionsPriority, #divFieldSuggestionsForm").forEach(AIAgentButton=>{     
+                let selected = settings('get','aiBehavior');
+                if (selected == "aiDisable") {
+                    AIAgentButton.style.display = "none"
+                }  
+                })
+            }
+            if (t.classList.contains("cke_button_label")){
+                document.querySelectorAll(".cke_button__revisetext").forEach(AIAgentButton=>{     
+                let selected = settings('get','aiBehavior');
+                if (selected == "aiDisable") {
+                    AIAgentButton.style.display = "none"
+                }  
+                })
+            }
             
             let module = t.querySelector("div > .tdx-dashboard__widget-container")
             if (module) {