diff --git a/contract_sale_generation/tests/test_contract_sale.py b/contract_sale_generation/tests/test_contract_sale.py
index 7f07fad3d..a5eaf6500 100644
--- a/contract_sale_generation/tests/test_contract_sale.py
+++ b/contract_sale_generation/tests/test_contract_sale.py
@@ -4,12 +4,12 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 from odoo.exceptions import ValidationError
-from odoo.tests.common import SavepointCase
+from odoo.tests.common import TransactionCase
 
 from .common import ContractSaleCommon, to_date
 
 
-class TestContractSale(ContractSaleCommon, SavepointCase):
+class TestContractSale(ContractSaleCommon, TransactionCase):
     def test_check_discount(self):
         with self.assertRaises(ValidationError):
             self.contract_line.write({"discount": 120})
diff --git a/contract_sale_generation/tests/test_contract_sale_recurrency.py b/contract_sale_generation/tests/test_contract_sale_recurrency.py
index 16572a55c..5c63a9e76 100644
--- a/contract_sale_generation/tests/test_contract_sale_recurrency.py
+++ b/contract_sale_generation/tests/test_contract_sale_recurrency.py
@@ -5,14 +5,14 @@ from freezegun.api import freeze_time
 
 from odoo import fields
 from odoo.tests import Form
-from odoo.tests.common import SavepointCase
+from odoo.tests.common import TransactionCase
 
 from .common import ContractSaleCommon
 
 today = "2020-01-15"
 
 
-class TestContractSale(ContractSaleCommon, SavepointCase):
+class TestContractSale(ContractSaleCommon, TransactionCase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()