diff --git a/product_usability/models/product_supplierinfo.py b/product_usability/models/product_supplierinfo.py
index 4f11d78..e3dacd0 100644
--- a/product_usability/models/product_supplierinfo.py
+++ b/product_usability/models/product_supplierinfo.py
@@ -9,4 +9,4 @@ from odoo import fields, models
 class ProductSupplierinfo(models.Model):
     _inherit = 'product.supplierinfo'
 
-    partner_id = fields.Many2one(domain=[('parent_id', '=', False)])
+    partner_id = fields.Many2one(domain=[('parent_id', '=', False)], ondelete='restrict')
diff --git a/product_usability/models/product_template.py b/product_usability/models/product_template.py
index 2e6a94e..e68ad56 100644
--- a/product_usability/models/product_template.py
+++ b/product_usability/models/product_template.py
@@ -21,6 +21,8 @@ class ProductTemplate(models.Model):
     # are only shown in the form view of product.template, not in the form
     # view of product.product
     name = fields.Char(tracking=10)
+    uom_id = fields.Many2one(tracking=15)
+    uom_po_id = fields.Many2one(tracking=15)
     barcode = fields.Char(tracking=20)
     default_code = fields.Char(tracking=30)
     categ_id = fields.Many2one(tracking=40)