Restore cleanup