small change to if statement