Fix bug in comments