Fix code generation.