diff --git a/addons/sale/tests/test_product_id_change.py b/addons/sale/tests/test_product_id_change.py
index cc94ad8eaf6..3864831e1e7 100644
--- a/addons/sale/tests/test_product_id_change.py
+++ b/addons/sale/tests/test_product_id_change.py
@@ -122,10 +122,10 @@ class TestOnchangeProductId(TransactionCase):
         computer_case = self.env.ref('product.product_product_16')
         computer_case.list_price = 100
         partner = self.res_partner_model.create(dict(name="George"))
-        categ_unit_id = self.ref('product.product_uom_unit')
+        categ_unit_id = self.ref('uom.product_uom_categ_unit')
         goup_discount_id = self.ref('sale.group_discount_per_so_line')
         self.env.user.write({'groups_id': [(4, goup_discount_id, 0)]})
-        new_uom = self.env['product.uom'].create({
+        new_uom = self.env['uom.uom'].create({
             'name': '10 units',
             'factor_inv': 10,
             'uom_type': 'bigger',
@@ -174,7 +174,6 @@ class TestOnchangeProductId(TransactionCase):
         computer_case = self.env.ref('product.product_product_16')
         computer_case.list_price = 100
         partner = self.res_partner_model.create(dict(name="George"))
-        categ_unit_id = self.ref('product.product_uom_unit')
         goup_discount_id = self.ref('sale.group_discount_per_so_line')
         self.env.user.write({'groups_id': [(4, goup_discount_id, 0)]})
 
@@ -225,7 +224,7 @@ class TestOnchangeProductId(TransactionCase):
         computer_case = self.env.ref('product.product_product_16')
         computer_case.list_price = 100
         partner = self.res_partner_model.create(dict(name="George"))
-        categ_unit_id = self.ref('product.product_uom_unit')
+        categ_unit_id = self.ref('uom.product_uom_categ_unit')
         other_currency = self.env['res.currency'].create({'name': 'other currency',
             'symbol': 'other'})
         self.env['res.currency.rate'].create({'name': '2018-07-11',
@@ -235,7 +234,7 @@ class TestOnchangeProductId(TransactionCase):
         self.env['res.currency.rate'].search(
             [('currency_id', '=', self.env.user.company_id.currency_id.id)]
         ).unlink()
-        new_uom = self.env['product.uom'].create({
+        new_uom = self.env['uom.uom'].create({
             'name': '10 units',
             'factor_inv': 10,
             'uom_type': 'bigger',