diff --git a/pool/tree/pool_server_test.go b/pool/tree/pool_server_test.go
index 7e3447f..edba93f 100644
--- a/pool/tree/pool_server_test.go
+++ b/pool/tree/pool_server_test.go
@@ -183,12 +183,15 @@ func sortServers(ctx context.Context, servers []*mockTreeServer, source *mockNet
 	}
 
 	// find servers based on public key and store pointers in res
+	index := 0
 	for i := range priorityNodes {
 		for j := range priorityNodes[i] {
 			key := priorityNodes[i][j].PublicKey()
 			for k := range servers {
 				if bytes.Equal(servers[k].key.PublicKey().Bytes(), key) {
-					res[i+j] = servers[k]
+					res[index] = servers[k]
+					index++
+					break
 				}
 			}
 		}