Fix code generation template