Minor change to if statement