Fix gen.py header