diff --git a/python_payload/st3m/goose.py b/python_payload/st3m/goose.py
index 25aaaf911e07ff8a037153c6d3825c471bc69a2b..5b678a896d4b43b70eda7735bd55de4381cddc3f 100644
--- a/python_payload/st3m/goose.py
+++ b/python_payload/st3m/goose.py
@@ -17,7 +17,7 @@ if TYPE_CHECKING:
     class ABCBase(metaclass=ABCMeta):
         pass
 
-    from typing import List, Optional, Tuple, Dict, Any, Callable, Iterator, Generator
+    from typing import List, Optional, Tuple, Dict, Any, Callable, Iterator, Generator, Union
     from enum import Enum
 else:
     # We're in CPython or Micropython.
@@ -40,6 +40,7 @@ else:
             Callable,
             Iterator,
             Generator,
+            Union,
         )
         from enum import Enum
     except ImportError:
@@ -52,6 +53,7 @@ else:
         Callable = None
         Iterator = None
         Generator = None
+        Union = None
 
         class Enum:
             pass
@@ -70,4 +72,5 @@ __all__ = [
     "Callable",
     "Iterator",
     "Generator",
+    "Union",
 ]