diff --git a/database/db_config.py b/database/db_config.py
index ee1673f..b75826d 100644
--- a/database/db_config.py
+++ b/database/db_config.py
@@ -1,9 +1,10 @@
 import os
 from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
 from sqlalchemy.pool import StaticPool
-from database.models import Base
 import logging
 
+from database.models.generic_model import Base
+
 logger = logging.getLogger(__name__)
 
 DATABASE_URL = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///./data/sage_dataven.db")