fix bug, add comment