diff --git a/setup.py b/setup.py
index 9b540124af..725e7996c8 100644
--- a/setup.py
+++ b/setup.py
@@ -30,6 +30,7 @@ setup(name='trytond_account_statement',
     package_dir={'trytond.modules.account_statement': '.'},
     packages=[
         'trytond.modules.account_statement',
+        'trytond.modules.account_statement.tests',
     ],
     package_data={
         'trytond.modules.account_statement': info.get('xml', []) \
@@ -58,4 +59,6 @@ setup(name='trytond_account_statement',
     [trytond.modules]
     account_statement = trytond.modules.account_statement
     """,
+    test_suite='tests',
+    test_loader='trytond.test_loader:Loader',
 )
diff --git a/tests/__init__.py b/tests/__init__.py
index bf3bc8c056..db046642a2 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,4 +1,4 @@
 #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_statement import *
+from test_account_statement import suite
diff --git a/tests/test_account_statement.py b/tests/test_account_statement.py
index d43a38cfc3..f3ea934a68 100644
--- a/tests/test_account_statement.py
+++ b/tests/test_account_statement.py
@@ -2,9 +2,6 @@
 #This file is part of Tryton.  The COPYRIGHT file at the top level of
 #this repository contains the full copyright notices and license terms.
 
-import logging
-logging.basicConfig(level=logging.FATAL)
-
 import sys, os
 DIR = os.path.abspath(os.path.normpath(os.path.join(__file__,
     '..', '..', '..', '..', '..', 'trytond')))