Corrected highlighting of comments on the same line as import