Fix get_features.py