Restore error handling