diff --git a/addons/sale/models/account_move_line.py b/addons/sale/models/account_move_line.py
index 43a0be53dba..9b8d08746d3 100644
--- a/addons/sale/models/account_move_line.py
+++ b/addons/sale/models/account_move_line.py
@@ -185,7 +185,7 @@ class AccountMoveLine(models.Model):
 
         if self.product_id.expense_policy == 'sales_price':
             return self.product_id.with_context(
-                partner=order.partner_id.id,
+                partner=order.partner_id,
                 date_order=order.date_order,
                 pricelist=order.pricelist_id.id,
                 uom=self.product_uom_id.id