fixed the wrong highlighting issue