Made code generator for python just work.