fix(trie): descendants count for clear prefix

ChainSafe · Jun 17, 2022 · 5d89380 · 5d89380
1 parent 223cfbb
commit 5d89380
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/trie/trie.go b/lib/trie/trie.go
@@ -877,10 +877,11 @@ func (t *Trie) clearPrefix(parent *Node, prefix []byte) (
 			return parent, nodesRemoved
 		}
 
-		nodesRemoved = 1
+		nodesRemoved = 1 + child.Descendants
 		copySettings := node.DefaultCopySettings
 		branch = t.prepBranchForMutation(branch, copySettings)
 		branch.Children[childIndex] = nil
+		branch.Descendants -= nodesRemoved
 		var branchChildMerged bool
 		newParent, branchChildMerged = handleDeletion(branch, prefix)
 		if branchChildMerged {