diff --git a/src/dscim/preprocessing/input_damages.py b/src/dscim/preprocessing/input_damages.py
index fda44343..04e81099 100644
--- a/src/dscim/preprocessing/input_damages.py
+++ b/src/dscim/preprocessing/input_damages.py
@@ -383,8 +383,8 @@ def process_batch(g):
     )
     batches = xr.where(np.isinf(batches), np.nan, batches)
 
-    batches.rename({"wc_reallocation": varname})[varname].to_zarr(
-        save_path, mode="a", consolidated=True
+    batches.rename({"wc_reallocation": varname})[varname].to_dataset().to_zarr(
+        store=save_path, mode="a", consolidated=True
     )