diff --git a/addons/stock_invoice_directly/test/stock_invoice_directly.yml b/addons/stock_invoice_directly/test/stock_invoice_directly.yml
index f07e2dbfa31..7c9626d1644 100644
--- a/addons/stock_invoice_directly/test/stock_invoice_directly.yml
+++ b/addons/stock_invoice_directly/test/stock_invoice_directly.yml
@@ -55,7 +55,7 @@
   !python {model: account.invoice}: |
     picking_obj = self.pool.get('stock.picking')
     picking = picking_obj.browse(cr, uid, [ref('stock_picking_out0')])
-    partner = picking[0].address_id.partner_id.id
+    partner = picking[0].address_id.id
     inv_ids = self.search(cr, uid, [('type','=','out_invoice'),('partner_id','=',partner)])
     assert inv_ids, 'No Invoice is generated!'