Address PR comment.