Fix bug in comments generator