Fix if/elseif/else mess