Minor change to IfThenElse code_gen