diff --git a/src/cloudcasting/utils.py b/src/cloudcasting/utils.py
index a495f7b..19ce71f 100644
--- a/src/cloudcasting/utils.py
+++ b/src/cloudcasting/utils.py
@@ -26,7 +26,7 @@
 def lon_lat_to_geostationary_area_coords(
     x: Sequence[float],
     y: Sequence[float],
-    xr_data: xr.Dataset,
+    xr_data: xr.Dataset | xr.DataArray,
 ) -> tuple[Sequence[float], Sequence[float]]:
     """Loads geostationary area and change from lon-lat to geostationaery coords
     Args: