bugfixes on codegen