fix end highlight