Fixes generated code.