diff --git a/Base/Python/slicer/util.py b/Base/Python/slicer/util.py
index 096c4b85e73..e1e3ff66680 100644
--- a/Base/Python/slicer/util.py
+++ b/Base/Python/slicer/util.py
@@ -2045,6 +2045,8 @@ def arrayFromSegmentBinaryLabelmap(segmentationNode, segmentId, referenceVolumeN
             raise RuntimeError("Export of segment failed.")
         narray = slicer.util.arrayFromVolume(labelmapVolumeNode)
     finally:
+        if labelmapVolumeNode.GetDisplayNode():
+            slicer.mrmlScene.RemoveNode(labelmapVolumeNode.GetDisplayNode().GetColorNode())
         slicer.mrmlScene.RemoveNode(labelmapVolumeNode)
 
     return narray