Fixed if-else statement