diff --git a/addons/activeblue_ai/models/ab_ai_mail.py b/addons/activeblue_ai/models/ab_ai_mail.py
index 5ade6fc..a14d438 100644
--- a/addons/activeblue_ai/models/ab_ai_mail.py
+++ b/addons/activeblue_ai/models/ab_ai_mail.py
@@ -13,8 +13,8 @@ def _strip_html(html: str) -> str:
     return _HTML_TAG.sub(' ', html or '').strip()
 
 
-class MailChannel(models.Model):
-    _inherit = 'mail.channel'
+class DiscussChannel(models.Model):
+    _inherit = 'discuss.channel'
 
     @api.model
     def _ai_bot_partner(self):