fix if-else statement