small bug fix for feature extraction