diff --git a/unstructured_inference/models/base.py b/unstructured_inference/models/base.py
index 14bb6570..59c3c704 100644
--- a/unstructured_inference/models/base.py
+++ b/unstructured_inference/models/base.py
@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 import json
 import os
-from typing import Dict, Optional, Tuple, Type, Union
+from typing import Dict, Optional, Tuple, Type
 
 from unstructured_inference.models.chipper import MODEL_TYPES as CHIPPER_MODEL_TYPES
 from unstructured_inference.models.chipper import UnstructuredChipperModel
@@ -25,7 +27,7 @@
 def get_default_model_mappings() -> (
     Tuple[
         Dict[str, Type[UnstructuredModel]],
-        Dict[str, Union[dict, LazyDict]],
+        Dict[str, dict | LazyDict],
     ]
 ):
     """default model mappings for models that are in `unstructured_inference` repo"""