Fix error handling restore