diff --git a/extensions/dicom-segmentation/src/components/SegmentationPanel/SegmentationPanel.js b/extensions/dicom-segmentation/src/components/SegmentationPanel/SegmentationPanel.js
index 00e193291..8bd61e3b7 100644
--- a/extensions/dicom-segmentation/src/components/SegmentationPanel/SegmentationPanel.js
+++ b/extensions/dicom-segmentation/src/components/SegmentationPanel/SegmentationPanel.js
@@ -403,13 +403,7 @@ const SegmentationPanel = ({
 
     let segmentsHidden = [];
     possibleLabelMaps3D.forEach(labelmap3D => {
-      if (isCornerstone()) {
-        labelmap3D.segmentsHidden[segmentNumber] = !isVisible;
-      }
-
-      if (isVTK()) {
-        onSegmentVisibilityChange(segmentNumber, isVisible);
-      }
+      labelmap3D.segmentsHidden[segmentNumber] = !isVisible;
 
       segmentsHidden = [
         ...new Set([...segmentsHidden, ...labelmap3D.segmentsHidden]),
@@ -420,6 +414,10 @@ const SegmentationPanel = ({
 
     refreshSegmentations();
     refreshViewports();
+
+    if (isVTK()) {
+      onSegmentVisibilityChange(segmentNumber, isVisible);
+    }
   };
 
   const getSegmentList = () => {
@@ -565,6 +563,10 @@ const SegmentationPanel = ({
   const onVisibilityChangeHandler = isVisible => {
     let segmentsHidden = [];
     state.segmentNumbers.forEach(segmentNumber => {
+      if (isVTK()) {
+        onSegmentVisibilityChange(segmentNumber, isVisible);
+      }
+
       /** Get all labelmaps with this segmentNumber (overlapping segments) */
       const { labelmaps3D } = getBrushStackState();
       const possibleLabelMaps3D = labelmaps3D.filter(({ labelmaps2D }) => {
@@ -574,14 +576,7 @@ const SegmentationPanel = ({
       });
 
       possibleLabelMaps3D.forEach(labelmap3D => {
-        if (isCornerstone()) {
-          labelmap3D.segmentsHidden[segmentNumber] = !isVisible;
-        }
-
-        if (isVTK()) {
-          onSegmentVisibilityChange(segmentNumber, isVisible);
-        }
-
+        labelmap3D.segmentsHidden[segmentNumber] = !isVisible;
         segmentsHidden = [
           ...new Set([...segmentsHidden, ...labelmap3D.segmentsHidden]),
         ];