Fix comments on PR