diff --git a/agency_swarm/util/streaming/gradio_event_handler.py b/agency_swarm/util/streaming/gradio_event_handler.py
index fe4ee3ce..2206a3d6 100644
--- a/agency_swarm/util/streaming/gradio_event_handler.py
+++ b/agency_swarm/util/streaming/gradio_event_handler.py
@@ -1,5 +1,5 @@
 import queue
-from typing import Type, override
+from typing import Type
 
 from openai.types.beta.threads import Message
 from openai.types.beta.threads.runs.run_step import RunStep
@@ -9,6 +9,7 @@
     FunctionToolCall,
     ToolCall,
 )
+from typing_extensions import override
 
 from agency_swarm.messages import MessageOutput
 from agency_swarm.util.streaming import (
diff --git a/agency_swarm/util/streaming/term_event_handler.py b/agency_swarm/util/streaming/term_event_handler.py
index 9b229a9e..8d1e7454 100644
--- a/agency_swarm/util/streaming/term_event_handler.py
+++ b/agency_swarm/util/streaming/term_event_handler.py
@@ -1,4 +1,4 @@
-from typing import Type, override
+from typing import Type
 
 from openai.types.beta.threads import Message
 from openai.types.beta.threads.runs import RunStep