fix(comments): handle comments on last lines