diff --git a/apps/api/app/api/routes/tts.py b/apps/api/app/api/routes/tts.py
index 56e73578f..7bb55d80b 100644
--- a/apps/api/app/api/routes/tts.py
+++ b/apps/api/app/api/routes/tts.py
@@ -22,7 +22,7 @@ from app.schemas.tts import (
     TTSSynthesizeRequest,
     TTSSynthesizeResponse,
 )
-from fastapi import APIRouter, Depends, HTTPException, Query, Response, WebSocket, status
+from fastapi import APIRouter, Depends, HTTPException, Query, Response, WebSocket, WebSocketDisconnect, status
 from sqlalchemy.orm import Session
 
 from packages.adapters.sqlalchemy_impl.tts_job_repository import (