diff --git a/server/chat/chat.py b/server/chat/chat.py
index 66dfd582..59728b43 100644
--- a/server/chat/chat.py
+++ b/server/chat/chat.py
@@ -3,7 +3,7 @@ import json
 from typing import AsyncIterable, List
 
 from fastapi import Body
-from fastapi.responses import StreamingResponse
+from sse_starlette import EventSourceResponse
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.messages import AIMessage, HumanMessage