diff --git a/main.py b/main.py
index 894744d..977b226 100644
--- a/main.py
+++ b/main.py
@@ -5,6 +5,7 @@ from starlette.staticfiles import StaticFiles
 from starlette.responses import FileResponse, RedirectResponse
 from fastapi.templating import Jinja2Templates
 from contextlib import asynccontextmanager
+from sqlalchemy.orm import Session
 
 import models
 from database import engine