Fixing the get_segmentation_id method.