Fixed botched if/else