fixed highlight problem