fix comments from PR