fixed the wrong highlighting issue
Merge pull request #171 from LI-HE/master

fixed the wrong highlighting issue