diff --git a/purchase_discount/tests/test_product_supplierinfo_discount.py b/purchase_discount/tests/test_product_supplierinfo_discount.py
index 089447eda03..6578a3acd48 100644
--- a/purchase_discount/tests/test_product_supplierinfo_discount.py
+++ b/purchase_discount/tests/test_product_supplierinfo_discount.py
@@ -2,11 +2,11 @@
 # Copyright 2019 Tecnativa - Pedro M. Baeza
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-import odoo.tests.common as common
 from odoo import fields
+from odoo.tests.common import TransactionCase
 
 
-class TestProductSupplierinfoDiscount(common.SavepointCase):
+class TestProductSupplierinfoDiscount(TransactionCase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()
diff --git a/purchase_discount/tests/test_purchase_discount.py b/purchase_discount/tests/test_purchase_discount.py
index 0c937912ca6..95f4edaad2f 100644
--- a/purchase_discount/tests/test_purchase_discount.py
+++ b/purchase_discount/tests/test_purchase_discount.py
@@ -2,11 +2,11 @@
 # Copyright 2015-2019 Tecnativa - Pedro M. Baeza
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-import odoo.tests.common as common
 from odoo import fields
+from odoo.tests.common import TransactionCase
 
 
-class TestPurchaseOrder(common.SavepointCase):
+class TestPurchaseOrder(TransactionCase):
     @classmethod
     def setUpClass(cls):
         super(TestPurchaseOrder, cls).setUpClass()