Fix minor comments on PR.