From 1826896e958d08aaf34660473ba30accc322179b Mon Sep 17 00:00:00 2001
From: Quentin McGaw <quentin.mcgaw@gmail.com>
Date: Wed, 29 Jun 2022 08:43:19 -0400
Subject: [PATCH] fix(trie): descendants count for clear prefix (#2606)

---
 lib/trie/trie.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/trie/trie.go b/lib/trie/trie.go
index 8f594e4180..bd26ceddfd 100644
--- 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 {