Fixing feature extractor bugs