Update _feature_gen.py

Quick little fix.