diff --git a/main.py b/main.py
index bbd88a5..80b292e 100644
--- a/main.py
+++ b/main.py
@@ -5,7 +5,6 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import datetime
-from sse_starlette import EventSourceResponse
 
 class Message(BaseModel):
     message: str
@@ -36,4 +35,4 @@ async def sse_endpoint(request: Request):
             yield f"data: Server time: {current_time}\n\n"
             await asyncio.sleep(1)
 
-    return EventSourceResponse(event_generator())
+    return StreamingResponse(event_generator(), media_type="text/event-stream")