diff --git a/src/PartSeg_smfish/segmentation.py b/src/PartSeg_smfish/segmentation.py
index e1777a7..3d5bdb5 100644
--- a/src/PartSeg_smfish/segmentation.py
+++ b/src/PartSeg_smfish/segmentation.py
@@ -740,12 +740,14 @@ def calculation_run(
         neigh, dist = calculate_distances_array(
             self.image.spacing, get_neigh(True)
         )
-        roi = euclidean_sprawl(
-            sprawl_area,
-            res.roi,
-            components_num,
-            neigh,
-            dist,
+        roi = self.image.fit_array_to_image(
+            euclidean_sprawl(
+                sprawl_area,
+                res.roi,
+                components_num,
+                neigh,
+                dist,
+            )
         )
 
         res2 = ROIExtractionResult(