Tweak to go with previous commit