diff --git a/tests/test_module.py b/tests/test_module.py
index e2de1f2..aca2ef5 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 SaleShopTradeInfoTestCase(CompanyTestMixin, ModuleTestCase):
+class SaleShopTradeInfoTestCase(SaleShopCompanyTestMixin, ModuleTestCase):
     'Test SaleShopTradeInfo module'
     module = 'sale_shop_trade_info'