Fix if/endif bug