Fix helper script for feature extraction