diff --git a/tests/__init__.py b/tests/__init__.py
index 29fbae418a..7ce46a9658 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,5 +1,8 @@
 # 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 .test_customs import suite
+try:
+    from trytond.modules.customs.tests.test_customs import suite
+except ImportError:
+    from .test_customs import suite
 
 __all__ = ['suite']