Fix the if statement