diff --git a/backend/app/backend_app.py b/backend/app/backend_app.py
index af5b44a..f3dfdc0 100644
--- a/backend/app/backend_app.py
+++ b/backend/app/backend_app.py
@@ -5,7 +5,7 @@
 from flask_migrate import Migrate
 from flask_jwt_extended import JWTManager
 from .routes import routes
-from models import db, User
+from .models import db, User
 from flask_cors import CORS
 
 DB_NAME = 'mygeneratordb'
diff --git a/backend/app/routes.py b/backend/app/routes.py
index b402eb1..c49d418 100644
--- a/backend/app/routes.py
+++ b/backend/app/routes.py
@@ -3,7 +3,7 @@
 from flask_jwt_extended import create_access_token, jwt_required, get_jwt_identity
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError
 from werkzeug.security import generate_password_hash, check_password_hash
-from models import User, Employee, ContractType, FinalContract, db
+from .models import User, Employee, ContractType, FinalContract, db
 from utils import generate_pdf, upload_to_s3
 import os
 import uuid
diff --git a/requirements.txt b/requirements.txt
index eeb142a..f22e0b1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -38,3 +38,4 @@ typing_extensions==4.12.2
 urllib3==1.26.20
 Werkzeug==2.2.3
 zipp==3.20.1
+pytest-flask==1.3.0