diff --git a/setup.py b/setup.py
index 1c9e7627c2..a5103df71f 100644
--- a/setup.py
+++ b/setup.py
@@ -30,6 +30,7 @@ setup(name='trytond_stock_supply',
     package_dir={'trytond.modules.stock_supply': '.'},
     packages=[
         'trytond.modules.stock_supply',
+        'trytond.modules.stock_supply.tests',
     ],
     package_data={
         'trytond.modules.stock_supply': info.get('xml', []) \
@@ -58,4 +59,6 @@ setup(name='trytond_stock_supply',
     [trytond.modules]
     stock_supply = trytond.modules.stock_supply
     """,
+    test_suite='tests',
+    test_loader='trytond.test_loader:Loader',
 )
diff --git a/tests/__init__.py b/tests/__init__.py
index 72fa326169..53c66b53ab 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_stock_supply import *
+from test_stock_supply import suite
diff --git a/tests/test_stock_supply.py b/tests/test_stock_supply.py
index 62df20bcbc..3c3e7cc493 100644
--- a/tests/test_stock_supply.py
+++ b/tests/test_stock_supply.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')))