diff --git a/__init__.py b/__init__.py
index 54438ee..152f152 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,14 +1,14 @@
 # The COPYRIGHT file at the top level of this repository contains the full
 # copyright notices and license terms.
 from trytond.pool import Pool
-from .asset import *
+from . import asset
 from . import party
 
 
 def register():
     Pool.register(
-        Asset,
-        AssetManager,
+        asset.Asset,
+        asset.AssetManager,
         module='asset_manager', type_='model')
     Pool.register(
         party.PartyReplace,
diff --git a/tests/__init__.py b/tests/__init__.py
index d77e821..1c2be3b 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,3 +1,5 @@
 # The COPYRIGHT file at the top level of this repository contains the full
 # copyright notices and license terms.
 from .test_asset_manager import suite
+
+__all__ = ['suite']