Small clean up for PR