Update codegen.py I'm not sure about it.