Fix highlighting of the current comment

[jumper047]

I'm not sure this is good solution, so please consider it as bug report if you want:) And thank you for that package, it's truly awesome! Love it, and hope you'll continue to develop it!