fix(trie): descendants count for clear prefix (#2606)

ChainSafe · Jun 29, 2022 · 1826896 · 1826896
1 parent 2a6fa83
commit 1826896
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 {