fixed feature extraction bug