fix bug with not including comment code