diff --git a/yaltik_crm_action/models/res_partner.py b/yaltik_crm_action/models/res_partner.py
index 63a07ad..e1fa8d8 100644
--- a/yaltik_crm_action/models/res_partner.py
+++ b/yaltik_crm_action/models/res_partner.py
@@ -50,12 +50,8 @@ class ResPartner(models.Model):
         """ Return action for active partner and its parent and childs """
         active_id = self._context.get('active_id')
         partner = self.env['res.partner'].browse(active_id)
-        if partner.parent_id:
-            partner_ids = [partner.parent_id.id] + partner.parent_id.child_ids.ids
-            domain = [('partner_id', 'in', partner_ids)]
-        elif partner.child_ids:
-            partner_ids = [partner.id] + partner.child_ids.ids
-            domain = [('partner_id', 'in', partner_ids)]
+        parent_id = partner.parent_id.id if partner.parent_id else partner.id
+        domain = [('partner_id', 'child_of', parent_id)]
 
         return {'type': 'ir.actions.act_window',
                 'name': _('Actions'),