Fix if/else