Fix if/else if/else error