minor fix
finish with generate.py