diff --git a/tests/__init__.py b/tests/__init__.py
index ad5c2233b5..8026d9a842 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -2,7 +2,8 @@
 # this repository contains the full copyright notices and license terms.
 
 try:
-    from trytond.modules.company.tests.test_company import suite
+    from trytond.modules.company.tests.test_company import (
+        suite, create_company, set_company)
 except ImportError:
     from .test_company import suite, create_company, set_company