fixed unit test for segmentation