Small improvement and add comment.