fix the if-else