diff --git a/raster_tools/rasterize.py b/raster_tools/rasterize.py
index 0859d6e..6a6c3c0 100644
--- a/raster_tools/rasterize.py
+++ b/raster_tools/rasterize.py
@@ -362,7 +362,7 @@ def _rasterize_spatial_matches(
     # Group by partition and iterate over the groups
     for ipart, grp in matches.groupby("part_idx"):
         # Get the vector partition
-        part = partitions[ipart]
+        part = partitions[ipart].to_delayed()[0]
         print(f"_rasterize_spatial_matches:\n{part.compute()}")
         # Iterate over the chunks that intersected the vector partition and
         # rasterize the partition to each intersecting chunk's grid