fix bug in FeatureExtractor