Edit the if statement