Fix bug
Add comment