Fix highlighting after undo assignment