Fixed a minior bug in comments