diff --git a/app/db/base.py b/app/db/base.py
index 8b9a4cd..dcedff3 100644
--- a/app/db/base.py
+++ b/app/db/base.py
@@ -1 +1,6 @@
-# Import all models for Alembic to detect
+# Import all models and base for Alembic to detect
+from app.db.base_class import Base  # noqa
+from app.models.todo import Todo  # noqa
+
+# Export Base for other modules to use
+__all__ = ["Base"]