Fix if/else.