Improve the comment