Fixes wrong highlighting problem.