Correcting for PR comments