diff --git a/tests/test_module.py b/tests/test_module.py
index d041178..7ab42c3 100644
--- a/tests/test_module.py
+++ b/tests/test_module.py
@@ -2,11 +2,11 @@
 # This file is part of Tryton.  The COPYRIGHT file at the top level of
 # this repository contains the full copyright notices and license terms.
 
-from trytond.modules.company.tests import CompanyTestMixin
+from trytond.modules.sale_shop.tests import SaleShopCompanyTestMixin
 from trytond.tests.test_tryton import ModuleTestCase
 
 
-class SaleRuleTestCase(CompanyTestMixin, ModuleTestCase):
+class SaleRuleTestCase(SaleShopCompanyTestMixin, ModuleTestCase):
     'Test SaleRule module'
     module = 'sale_rule'
     extras = ['sale_pos', 'sale_discount']