fixed a bug with wrong comments

this will prevent invalid comments being generated
Merge pull request #9 from JohnHeroHD/master

fixed a bug with wrong comments being generated