diff --git a/tests/__init__.py b/tests/__init__.py
index fcab20999e..e8604144ae 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,6 +1,9 @@
 # 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_account_invoice import suite, set_invoice_sequences
+try:
+    from trytond.modules.account_invoice.tests.test_account_invoice import suite
+except ImportError:
+    from .test_account_invoice import suite, set_invoice_sequences
 
 __all__ = ['suite', 'set_invoice_sequences']