[patch] Lazy highlighting improvement