diff --git a/openeo-geotrellis/src/main/scala/org/openeo/geotrellis/OpenEOProcesses.scala b/openeo-geotrellis/src/main/scala/org/openeo/geotrellis/OpenEOProcesses.scala
index b0445ef8..ba2ff3fd 100644
--- a/openeo-geotrellis/src/main/scala/org/openeo/geotrellis/OpenEOProcesses.scala
+++ b/openeo-geotrellis/src/main/scala/org/openeo/geotrellis/OpenEOProcesses.scala
@@ -1259,7 +1259,7 @@ class OpenEOProcesses extends Serializable {
       MultibandTile(filter.createMask(tile.tile).crop(originalBounds))
     })
     val updatedMetadata = datacube.metadata.copy(cellType = BitCellType)
-    ContextRDD(mask, updatedMetadata)
+    ContextRDD(new ShuffledRDD[SpaceTimeKey, MultibandTile,MultibandTile](mask,mask.partitioner.get), updatedMetadata)
   }
 
   def mergeTiles(tiles: MultibandTileLayerRDD[SpaceTimeKey]): MultibandTileLayerRDD[SpaceTimeKey] = {