fix the highlight styles lose