Restore image error handling