Tweak the if statement a bit