diff --git a/jupyter_ai_tutor/handlers.py b/jupyter_ai_tutor/handlers.py
index 4f62e1c..0c3f301 100644
--- a/jupyter_ai_tutor/handlers.py
+++ b/jupyter_ai_tutor/handlers.py
@@ -101,7 +101,7 @@ async def post(self):
                     if debug_mode:
                         accumulated_response += text
                     self.write(f"data: {json.dumps({'text': text})}\n\n")
-                    self.flush()
+                    await self.flush()
 
             if debug_mode and answer_file:
                 try:
@@ -124,13 +124,13 @@ async def post(self):
             self.log.exception("Error during tutor LLM call")
             try:
                 self.write(f"data: {json.dumps({'error': str(e)})}\n\n")
-                self.flush()
+                await self.flush()
             except StreamClosedError:
                 return
 
         try:
             self.write("data: [DONE]\n\n")
-            self.flush()
+            await self.flush()
             self.finish()
         except StreamClosedError:
             pass