diff --git a/main.py b/main.py
index a682517..aa40322 100644
--- a/main.py
+++ b/main.py
@@ -4,6 +4,7 @@ from starlette.middleware.sessions import SessionMiddleware
 from starlette.staticfiles import StaticFiles 
 from starlette.responses import FileResponse, RedirectResponse
 from fastapi.templating import Jinja2Templates
+from starlette.websockets import WebSocket
 from contextlib import asynccontextmanager
 from sqlalchemy.orm import Session