diff --git a/purchase_stock_ux/models/purchase_order_line.py b/purchase_stock_ux/models/purchase_order_line.py
index 134ccbbd..bbd9bcf4 100644
--- a/purchase_stock_ux/models/purchase_order_line.py
+++ b/purchase_stock_ux/models/purchase_order_line.py
@@ -159,7 +159,7 @@ def _compute_qty_returned(self):
             for move in line.move_ids.filtered(
                 lambda m: m.state == "done" and m.location_id.usage != "supplier" and m.to_refund
             ):
-                qty += move.product_uom._compute_quantity(move.product_uom_qty, line.product_uom)
+                qty += move.product_uom._compute_quantity(move.product_uom_qty, line.product_uom_id)
             line.qty_returned = qty
 
     # Overwrite the origin method to introduce the qty_on_voucher