diff --git a/party.py b/party.py
index 704ff1e..29f4fd7 100644
--- a/party.py
+++ b/party.py
@@ -6,8 +6,6 @@ from trytond.tools.multivalue import migrate_property
 from trytond.modules.company.model import CompanyValueMixin
 
 
-__all__ = ['Party', 'PartyInvoiceTo']
-
 parties_names = ['customer_to_invoice', 'supplier_to_invoice']
 
 
diff --git a/setup.py b/setup.py
index 8899cb4..3535b26 100644
--- a/setup.py
+++ b/setup.py
@@ -121,7 +121,4 @@ setup(name=name,
     [trytond.modules]
     account_invoice_party_invoice_to = trytond.modules.account_invoice_party_invoice_to
     """,
-    test_suite='tests',
-    test_loader='trytond.test_loader:Loader',
-    tests_require=tests_require,
     )
diff --git a/tests/__init__.py b/tests/__init__.py
index 4c9024c..17f4e91 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,3 +1,2 @@
 # The COPYRIGHT file at the top level of this repository contains the full
 # copyright notices and license terms.
-from .test_account_invoice import suite
diff --git a/tests/test_account_invoice.py b/tests/test_module.py
similarity index 83%
rename from tests/test_account_invoice.py
rename to tests/test_module.py
index e5ed93c..7d42f93 100644
--- a/tests/test_account_invoice.py
+++ b/tests/test_module.py
@@ -1,8 +1,6 @@
 # The COPYRIGHT file at the top level of this repository contains the full
 # copyright notices and license terms.
-import unittest
 from trytond.pool import Pool
-import trytond.tests.test_tryton
 from trytond.tests.test_tryton import ModuleTestCase, with_transaction
 
 
@@ -32,7 +30,4 @@ class TestCase(ModuleTestCase):
         self.assertEqual(party.supplier_to_invoice, party2)
 
 
-def suite():
-    suite = trytond.tests.test_tryton.suite()
-    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestCase))
-    return suite
+del ModuleTestCase
diff --git a/tests/test_scenario.py b/tests/test_scenario.py
new file mode 100644
index 0000000..10388fe
--- /dev/null
+++ b/tests/test_scenario.py
@@ -0,0 +1,7 @@
+# The COPYRIGHT file at the top level of this repository contains
+# the full copyright notices and license terms.
+from trytond.tests.test_tryton import load_doc_tests
+
+
+def load_tests(*args, **kwargs):
+    return load_doc_tests(__name__, __file__, *args, **kwargs)
diff --git a/tryton.cfg b/tryton.cfg
index 06ff609..80cf01a 100644
--- a/tryton.cfg
+++ b/tryton.cfg
@@ -1,5 +1,5 @@
 [tryton]
-version=6.0.0
+version=6.4.0
 depends:
   account_invoice
   ir