diff --git a/app_chatgpt/models/mail_channel.py b/app_chatgpt/models/mail_channel.py
index 3c8ce56f..d789ca2f 100644
--- a/app_chatgpt/models/mail_channel.py
+++ b/app_chatgpt/models/mail_channel.py
@@ -61,8 +61,8 @@ class Channel(models.Model):
         if channel_type == 'chat':
             channel_partner_ids = self.channel_partner_ids
             to_partner_id = channel_partner_ids - message.author_id
-            user_id = to_partner_id.related_user_id
-            if user_id and to_partner_id.gpt_id:
+            user_id = to_partner_id.mapped('user_ids').filtered(lambda r: r.gpt_id)[:1]
+            if user_id:
                 gpt_policy = user_id.gpt_policy
                 gpt_wl_users = user_id.gpt_wl_users
                 is_allow = message.create_uid.id in gpt_wl_users.ids
@@ -75,8 +75,8 @@ class Channel(models.Model):
             if partner_ids:
                 partners = self.env['res.partner'].search([('id', 'in', partner_ids)])
                 # user_id = user has binded gpt robot
-                user_id = partners.mapped('related_user_id').filtered(lambda r: r.gpt_id)[:1]
-                if user_id and user_id.gpt_id:
+                user_id = partners.mapped('user_ids').filtered(lambda r: r.gpt_id)[:1]
+                if user_id:
                     gpt_policy = user_id.gpt_policy
                     gpt_wl_users = user_id.gpt_wl_users
                     is_allow = message.create_uid.id in gpt_wl_users.ids