diff --git a/src/openeo_processes/cubes.py b/src/openeo_processes/cubes.py
index 88a4835b..af19f3cb 100644
--- a/src/openeo_processes/cubes.py
+++ b/src/openeo_processes/cubes.py
@@ -235,9 +235,12 @@ def exec_xar(data, output_filepath='out', format='GTiff', options={}, write_prod
         data = data.fillna(-9999)
         data.attrs["nodata"] = -9999
         # Convert data array to data set, keeping a nice format on the bands.
-        data = data.to_dataset(
-            dim='bands'
-        )
+        if 'bands' in data.dims:
+            data = data.to_dataset(
+                dim='bands'
+            )
+        else:
+            data = data.to_dataset()
 
         if "crs" not in data.attrs:
             first_data_var = data.data_vars[list(data.data_vars.keys())[0]]