diff --git a/src/EdgeGPT.py b/src/EdgeGPT.py
index e797995ce..c900d49f8 100644
--- a/src/EdgeGPT.py
+++ b/src/EdgeGPT.py
@@ -152,7 +152,7 @@ def __init__(self, conversation: Conversation) -> None:
             conversation_id=conversation.struct["conversationId"],
         )
 
-    async def ask_stream(self, prompt: str) -> Generator[str]:
+    async def ask_stream(self, prompt: str) -> Generator[str, None, None]:
         """
         Ask a question to the bot
         """
@@ -220,7 +220,7 @@ async def ask(self, prompt: str) -> dict:
             if final:
                 return response
 
-    async def ask_stream(self, prompt: str) -> Generator[str]:
+    async def ask_stream(self, prompt: str) -> Generator[str, None, None]:
         """
         Ask a question to the bot
         """