small bug in segmentation/init.py