Adjust if statement