Fixed bug in comments