fixed the behaviour for some types of comments