Fixing code generation check.