Merge c5b77c8 into 22852aa

celestiaorg · Feb 20, 2021 · c10fb3c · c10fb3c
2 parents 22852aa + c5b77c8
commit c10fb3c
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 newRoot, err
 	}
+	smt.SetRoot(newRoot)
 	return newRoot, err
 }