fix on the extraction model