Update test for feature extraction