fix highlight problem