Fix if-then-else one liner