Added restore helper