diff --git a/purchase_product_restrict_supplier/models/purchase_order_line.py b/purchase_product_restrict_supplier/models/purchase_order_line.py
index f7526ed..9134906 100644
--- a/purchase_product_restrict_supplier/models/purchase_order_line.py
+++ b/purchase_product_restrict_supplier/models/purchase_order_line.py
@@ -11,15 +11,13 @@ _logger = logging.getLogger(__name__)
 class PurchaseOrderLine(models.Model):
     _inherit = 'purchase.order.line'
 
-    @api.multi
+    @api.onchange('product_id')
     def onchange_product_id(self):
-        _logger.warning("begin onchange_product_id")
         result = super(PurchaseOrderLine, self).onchange_product_id()
-        supplier_infos = self.env['product.supplierinfo'].search([('name', '=', self.partner_id.comercial)])
+        supplier_infos = self.env['product.supplierinfo'].search([('name', '=', self.partner_id.name)])
         product_ids = self.env['product.product']
         for supplier_info in supplier_infos:
              product_ids += supplier_info.product_tmpl_id.product_variant_ids
-
         result.update({'domain': {'product_id': [('id', 'in', product_ids.ids)]}})
 
         return result