diff --git a/src/compub/db/__init__.py b/src/compub/db/__init__.py
index 3b794e3..eb71b46 100644
--- a/src/compub/db/__init__.py
+++ b/src/compub/db/__init__.py
@@ -1 +1,2 @@
-from .base import LocalAsyncSession, ORMBase, engine
+from .geography import *
+from .companies import *
diff --git a/src/compub/db/companies.py b/src/compub/db/companies.py
index 077be5e..7158a52 100644
--- a/src/compub/db/companies.py
+++ b/src/compub/db/companies.py
@@ -17,6 +17,15 @@ from compub.utils import multi_max
 from .base import AbstractBase, ORMBase
 
 
+__all__ = [
+    'LegalForm',
+    'LegalFormSubcategory',
+    'Integer',
+    'Company',
+    'CompanyName',
+]
+
+
 class LegalForm(AbstractBase):
     __tablename__ = 'legal_form'
 
diff --git a/src/compub/db/geography.py b/src/compub/db/geography.py
index 35c0d29..f40af5f 100644
--- a/src/compub/db/geography.py
+++ b/src/compub/db/geography.py
@@ -8,6 +8,14 @@ from sqlalchemy_utils.types import CountryType, UUIDType
 from .base import AbstractBase
 
 
+__all__ = [
+    'StateProvince',
+    'City',
+    'Street',
+    'Address',
+]
+
+
 class StateProvince(AbstractBase):
     __tablename__ = 'state_province'