Fix multi plane segmentation for only one plane