diff --git a/src/core/settings.py b/src/core/settings.py
index 3150467..e467efd 100644
--- a/src/core/settings.py
+++ b/src/core/settings.py
@@ -4,7 +4,6 @@ from pydantic import BaseModel
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 BASE_DIR = Path(__file__).parent.parent
-print(BASE_DIR)
 DB_PATH = BASE_DIR / "db/taskncoffee.db"
 
 
diff --git a/src/migrations/env.py b/src/migrations/env.py
index ff19274..2bfab37 100644
--- a/src/migrations/env.py
+++ b/src/migrations/env.py
@@ -5,7 +5,7 @@ from sqlalchemy import pool
 
 from alembic import context
 
-from src.db.database import Base
+from src.core.database import Base
 from src.models import *  # noqa
 
 # this is the Alembic Config object, which provides