diff --git a/detectree2/models/outputs.py b/detectree2/models/outputs.py
index 6a9f7f60..b9a34552 100644
--- a/detectree2/models/outputs.py
+++ b/detectree2/models/outputs.py
@@ -9,7 +9,7 @@
 import re
 from http.client import REQUEST_URI_TOO_LONG  # noqa: F401
 from pathlib import Path
-from typing import Optional, Dict, Union, List, Any, TypedDict
+from typing import Any, Dict, List, Optional, TypedDict
 
 import cv2
 import geopandas as gpd