if-then-else fix