diff --git a/python/tvm/tir/schedule/_type_checker.py b/python/tvm/tir/schedule/_type_checker.py
index 12ce1ebc1f92..cb2d6446b3ef 100644
--- a/python/tvm/tir/schedule/_type_checker.py
+++ b/python/tvm/tir/schedule/_type_checker.py
@@ -27,6 +27,19 @@ def _is_none_type(type_: Any) -> bool:
     return type_ is None or type_ is type(None)
 
 
+def _get_subtypes(type_: Any) -> Any:
+    # TODO(@tvm-team): This is hot fix to support subtle difference between python versions
+    #                  Would be nice to find a better way if possible
+    if hasattr(typing, "_SpecialGenericAlias"):
+        if hasattr(typing, "get_args"):
+            subtypes = typing.get_args(type_)  # type: ignore
+        else:
+            subtypes = type_.__args__
+    else:
+        subtypes = type_.__args__
+    return subtypes
+
+
 if hasattr(typing, "_GenericAlias"):
     # For python versions 3.7 onward, check the __origin__ attribute.
 
@@ -64,10 +77,7 @@ def dict_(type_: Any) -> Any:
         @staticmethod
         def tuple_(type_: Any) -> Optional[List[type]]:
             if _Subtype._origin(type_) is tuple:
-                if hasattr(typing, "get_args"):
-                    subtypes = typing.get_args(type_)  # type: ignore
-                else:
-                    subtypes = type_.__args__
+                subtypes = _get_subtypes(type_)
                 return subtypes
             return None
 
@@ -76,10 +86,7 @@ def optional(  # pylint: disable=missing-function-docstring
             type_: Any,
         ) -> Optional[List[type]]:
             if _Subtype._origin(type_) is Union:
-                if hasattr(typing, "get_args"):
-                    subtypes = typing.get_args(type_)  # type: ignore
-                else:
-                    subtypes = type_.__args__
+                subtypes = _get_subtypes(type_)
                 if len(subtypes) == 2 and _is_none_type(subtypes[1]):
                     return [subtypes[0]]
             return None
@@ -87,10 +94,7 @@ def optional(  # pylint: disable=missing-function-docstring
         @staticmethod
         def union(type_: Any) -> Optional[List[type]]:  # pylint: disable=missing-function-docstring
             if _Subtype._origin(type_) is Union:
-                if hasattr(typing, "get_args"):
-                    subtypes = typing.get_args(type_)  # type: ignore
-                else:
-                    subtypes = type_.__args__
+                subtypes = _get_subtypes(type_)
                 if len(subtypes) != 2 or not _is_none_type(subtypes[1]):
                     return list(subtypes)
             return None
@@ -98,10 +102,7 @@ def union(type_: Any) -> Optional[List[type]]:  # pylint: disable=missing-functi
         @staticmethod
         def callable(type_: Any) -> Optional[List[type]]:
             if _Subtype._origin(type_) is collections.abc.Callable:
-                if hasattr(typing, "get_args") and not type_._special:
-                    subtypes = typing.get_args(type_)  # type: ignore
-                else:
-                    subtypes = type_.__args__
+                subtypes = _get_subtypes(type_)
                 return subtypes
             return None