diff --git a/app/db/db.py b/app/db/db.py
index e89762b..401bdd6 100644
--- a/app/db/db.py
+++ b/app/db/db.py
@@ -9,7 +9,7 @@ from sqlalchemy.orm import Session, sessionmaker
 Base = declarative_base()
 
 # Database configuration
-DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///investors_2.db")
+DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///investors.db")
 
 # Create engine
 engine = create_engine(DATABASE_URL, echo=False)
diff --git a/app/db/models.py b/app/db/models.py
index 4d2ce86..8ab6287 100644
--- a/app/db/models.py
+++ b/app/db/models.py
@@ -16,7 +16,6 @@ class InvestmentStage(enum.Enum):
     GROWTH = "GROWTH"
     LATE_STAGE = "LATE_STAGE"
 
-
 # Association table for many-to-many relationship between investors and companies
 investor_company_association = Table(
     "investor_companies",
diff --git a/app/services/llm_parser.py b/app/services/llm_parser.py
index 4465251..12fca8b 100644
--- a/app/services/llm_parser.py
+++ b/app/services/llm_parser.py
@@ -9,7 +9,7 @@ from dotenv import load_dotenv
 from openai import OpenAI
 
 from db import get_session, init_database
-from schema import CSVRow, Investor
+from py_schemas import CSVRow, Investor
 
 # Load environment variables
 load_dotenv()