diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py
index 79048a8d..65e9fea9 100644
--- a/vision_agent/utils/execute.py
+++ b/vision_agent/utils/execute.py
@@ -706,7 +706,7 @@ def new_instance(
         return instance
 
 
-def _get_e2b_env() -> Dict[str, str] | None:
+def _get_e2b_env() -> Union[Dict[str, str], None]:
     openai_api_key = os.getenv("OPENAI_API_KEY", "")
     anthropic_api_key = os.getenv("ANTHROPIC_API_KEY", "")
     if openai_api_key or anthropic_api_key: