diff --git a/npiai/tools/email_organizer/app.py b/npiai/tools/email_organizer/app.py
index 14ca782..34324d7 100644
--- a/npiai/tools/email_organizer/app.py
+++ b/npiai/tools/email_organizer/app.py
@@ -167,7 +167,7 @@ async def summarize_stream(
         output_columns: List[Column],
         with_pdf_attachments: bool = False,
         concurrency: int = 1,
-    ):
+    ) -> AsyncGenerator[EmailSummary, None]:
         """
         Summarize emails into a table.
 
@@ -181,7 +181,7 @@ async def summarize_stream(
         lock = asyncio.Lock()
         index = 0
 
-        async def summarize_email(results_queue: asyncio.Queue[FilterResult]):
+        async def summarize_email(results_queue: asyncio.Queue[EmailSummary]):
             nonlocal index
             async with lock:
                 if index >= len(email_or_id_list):