if/else fix