diff --git a/addons/purchase/test/average_price.yml b/addons/purchase/test/average_price.yml
index 0e2f0ff1240..5d5d6e33943 100644
--- a/addons/purchase/test/average_price.yml
+++ b/addons/purchase/test/average_price.yml
@@ -23,7 +23,7 @@
     order_line:
       - product_id: product_average_icecream
         product_qty: 10.0
-        product_uom: product.product_uom_categ_kgm
+        product_uom: product.product_uom_kgm
         price_unit: 60.0
         name: 'Average Ice Cream'
 - 
@@ -58,7 +58,7 @@
     order_line:
       - product_id: product_average_icecream
         product_qty: 30.0
-        product_uom: product.product_uom_categ_kgm
+        product_uom: product.product_uom_kgm
         price_unit: 80.0
         name: 'Average Ice Cream'
 -