From b37a7480a39aaf3cadb113aad3bf21b4eff52818 Mon Sep 17 00:00:00 2001
From: mcrakhman <rakhmanov.m@gmail.com>
Date: Thu, 26 Jan 2023 12:55:34 +0100
Subject: [PATCH] Set heads in tree storage

---
 commonspace/object/tree/exporter/treeexporter.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/commonspace/object/tree/exporter/treeexporter.go b/commonspace/object/tree/exporter/treeexporter.go
index 01d41f96..3dabd605 100644
--- a/commonspace/object/tree/exporter/treeexporter.go
+++ b/commonspace/object/tree/exporter/treeexporter.go
@@ -73,5 +73,8 @@ func (t *treeExporter) ExportUnencrypted(tree objecttree.ReadableObjectTree) (er
 		err = putStorage(change)
 		return err == nil
 	})
-	return err
+	if err != nil {
+		return
+	}
+	return treeStorage.SetHeads(tree.Heads())
 }