fix model for code generation