Repair custom segue class references