diff --git a/app/models/__init__.py b/app/models/__init__.py
index f23242c..cae56fe 100644
--- a/app/models/__init__.py
+++ b/app/models/__init__.py
@@ -1,4 +1,8 @@
+from app.database import Base
+
 from app.models.item import Item
 from app.models.category import Category
 from app.models.supplier import Supplier
-from app.models.stock_movement import StockMovement, MovementType
\ No newline at end of file
+from app.models.stock_movement import StockMovement, MovementType
+
+__all__ = ["Base", "Item", "Category", "Supplier", "StockMovement", "MovementType"]
\ No newline at end of file