diff --git a/packages/tools/src/eventListeners/segmentation/labelmap/onLabelmapSegmentationDataModified.ts b/packages/tools/src/eventListeners/segmentation/labelmap/onLabelmapSegmentationDataModified.ts
index 11525a7b6..a9f7ee1b9 100644
--- a/packages/tools/src/eventListeners/segmentation/labelmap/onLabelmapSegmentationDataModified.ts
+++ b/packages/tools/src/eventListeners/segmentation/labelmap/onLabelmapSegmentationDataModified.ts
@@ -152,7 +152,7 @@ function performStackLabelmapUpdate({ viewportIds, segmentationId }) {
             segmentationId
           );
 
-        const segmentationImage = cache.getImage(currentSegmentationImageId);
+        const segmentationImage = cache.getImage(currentSegmentationImageId[0]);
         segImageData.modified();
 
         // update the cache with the new image data