add restore tests