diff --git a/pkg/node/peer.go b/pkg/node/peer.go
index 3265c75..e897b31 100644
--- a/pkg/node/peer.go
+++ b/pkg/node/peer.go
@@ -556,8 +556,7 @@ func (r *PeerRegistry) Count() int {
 	return len(r.peers)
 }
 
-// rebuildKDTree rebuilds the KD-tree from current peers.
-// Must be called with lock held.
+// r.rebuildKDTree() // called after AddPeer, UpdatePeer, RemovePeer, UpdateMetrics — lock must be held
 func (r *PeerRegistry) rebuildKDTree() {
 	if len(r.peers) == 0 {
 		r.kdTree = nil