fixed highlight bug