fix bug and update comment