diff --git a/account_payment_purchase/models/purchase_order.py b/account_payment_purchase/models/purchase_order.py
index 6bb4f92af..580ecb62d 100644
--- a/account_payment_purchase/models/purchase_order.py
+++ b/account_payment_purchase/models/purchase_order.py
@@ -29,8 +29,9 @@ class PurchaseOrder(models.Model):
 
     @api.onchange("partner_id", "company_id")
     def onchange_partner_id(self):
-        ret = super(PurchaseOrder, self).onchange_partner_id()
+        res = super(PurchaseOrder, self).onchange_partner_id()
         if self.partner_id:
+            self = self.with_company(self.company_id)
             self.supplier_partner_bank_id = self._get_default_supplier_partner_bank(
                 self.partner_id
             )
@@ -38,4 +39,4 @@ class PurchaseOrder(models.Model):
         else:
             self.supplier_partner_bank_id = False
             self.payment_mode_id = False
-        return ret
+        return res