fix highlighting problem