fix highlight bug