Fix if-else logic.