Fix highlighting when using fuzzy