diff --git a/setup/utils/ui.py b/setup/utils/ui.py
index 65545c3..7356cba 100644
--- a/setup/utils/ui.py
+++ b/setup/utils/ui.py
@@ -6,7 +6,7 @@ Cross-platform console UI with colors and progress indication
 import sys
 import time
 import shutil
-from typing import List, Optional, Any, Dict
+from typing import List, Optional, Any, Dict, Union
 from enum import Enum
 
 # Try to import colorama for cross-platform color support
@@ -167,7 +167,7 @@ class Menu:
         self.multi_select = multi_select
         self.selected = set() if multi_select else None
         
-    def display(self) -> int | List[int]:
+    def display(self) -> Union[int, List[int]]:
         """
         Display menu and get user selection
         
@@ -431,4 +431,4 @@ def truncate_text(text: str, max_length: int, suffix: str = "...") -> str:
     if len(text) <= max_length:
         return text
     
-    return text[:max_length - len(suffix)] + suffix
\ No newline at end of file
+    return text[:max_length - len(suffix)] + suffix