diff --git a/mail_restrict_follower_selection/models/mail_followers.py b/mail_restrict_follower_selection/models/mail_followers.py
index 06a9aaaf4..3576e1552 100644
--- a/mail_restrict_follower_selection/models/mail_followers.py
+++ b/mail_restrict_follower_selection/models/mail_followers.py
@@ -35,10 +35,11 @@ class MailFollowers(models.Model):
                 existing_policy=existing_policy,
             )
         domain = self.env[
-            "mail.wizard.invite"
-        ]._mail_restrict_follower_selection_get_domain()
-        partners = self.env["res.partner"].search(
-            [("id", "in", partner_ids)] + safe_eval(domain)
+            'mail.wizard.invite'
+        ]._mail_restrict_follower_selection_get_domain(res_model=res_model)
+        partners = self.env['res.partner'].search(
+            [('id', 'in', partner_ids)] +
+            safe_eval(domain)
         )
         _res_ids = res_ids.copy() or [0]
         new, update = super()._add_followers(
diff --git a/mail_restrict_follower_selection/models/mail_wizard_invite.py b/mail_restrict_follower_selection/models/mail_wizard_invite.py
index e04c486cf..0d579f785 100644
--- a/mail_restrict_follower_selection/models/mail_wizard_invite.py
+++ b/mail_restrict_follower_selection/models/mail_wizard_invite.py
@@ -11,19 +11,15 @@ class MailWizardInvite(models.TransientModel):
     _inherit = "mail.wizard.invite"
 
     @api.model
-    def _mail_restrict_follower_selection_get_domain(self):
-        parameter_name = "mail_restrict_follower_selection.domain"
-        return (
-            self.env["ir.config_parameter"]
-            .sudo()
-            .get_param(
-                "{}.{}".format(
-                    parameter_name, self.env.context.get("default_res_model")
-                ),
-                self.env["ir.config_parameter"]
-                .sudo()
-                .get_param(parameter_name, default="[]"),
-            )
+    def _mail_restrict_follower_selection_get_domain(self, res_model=None):
+        if not res_model:
+            res_model = self.env.context.get('default_res_model')
+        parameter_name = 'mail_restrict_follower_selection.domain'
+        return self.env['ir.config_parameter'].sudo().get_param(
+            "{0}.{1}".format(parameter_name,
+                             res_model),
+            self.env['ir.config_parameter'].sudo().get_param(
+                parameter_name, default='[]')
         )
 
     @api.model