fix bug with comments