Add restore function reference