Fix highlighting after search in code editor