Fixed bug in segmentation