Merge be618ae into 2b025d3

celestiaorg · Feb 27, 2021 · 8f7afd5 · 8f7afd5
2 parents 2b025d3 + be618ae
commit 8f7afd5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/smt.go b/smt.go
@@ -138,9 +138,10 @@ func (smt *SparseMerkleTree) HasForRoot(key, root []byte) (bool, error) {
 // Update sets a new value for a key in the tree, and sets and returns the new root of the tree.
 func (smt *SparseMerkleTree) Update(key []byte, value []byte) ([]byte, error) {
 	newRoot, err := smt.UpdateForRoot(key, value, smt.Root())
-	if err == nil {
-		smt.SetRoot(newRoot)
+	if err != nil {
+		return nil, err
 	}
+	smt.SetRoot(newRoot)
 	return newRoot, err
 }