Update comment based on PR discussion.