diff --git a/__init__.py b/__init__.py
index a2fd1d8..9302c18 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,11 +1,9 @@
 # The COPYRIGHT file at the top level of this repository contains the full
 # copyright notices and license terms.
-
 from trytond.pool import Pool
-from .shop import *
-
+from . import shop
 
 def register():
     Pool.register(
-        Shop,
+        shop.Shop,
         module='sale_shop_trade_info', type_='model')
diff --git a/shop.py b/shop.py
index 79318cd..d2df339 100644
--- a/shop.py
+++ b/shop.py
@@ -4,8 +4,6 @@
 from trytond.model import fields
 from trytond.pool import PoolMeta
 
-__all__ = ['Shop']
-
 
 class Shop(metaclass=PoolMeta):
     __name__ = 'sale.shop'
diff --git a/tests/__init__.py b/tests/__init__.py
index 3ef2501..a668e2d 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,4 +1,5 @@
 # The COPYRIGHT file at the top level of this repository contains the full
 # copyright notices and license terms.
-
 from .test_sale_shop_trade_info import suite
+
+__all__ = ['suite']