Fix the bug and update the comment