Change the Segmentator#extract API