diff --git a/tests/test_purchase.py b/tests/test_purchase.py
index c7bda66e52..ff35a0f147 100644
--- a/tests/test_purchase.py
+++ b/tests/test_purchase.py
@@ -10,11 +10,12 @@ from trytond.tests.test_tryton import doctest_teardown, doctest_checker
 from trytond.transaction import Transaction
 from trytond.pool import Pool
 
-from trytond.modules.company.tests import create_company, set_company
+from trytond.modules.company.tests import (
+    create_company, set_company, PartyCompanyCheckEraseMixin)
 from trytond.modules.account.tests import create_chart
 
 
-class PurchaseTestCase(ModuleTestCase):
+class PurchaseTestCase(PartyCompanyCheckEraseMixin, ModuleTestCase):
     'Test Purchase module'
     module = 'purchase'