diff --git a/shotover-proxy/src/transforms/cassandra/sink_cluster/mod.rs b/shotover-proxy/src/transforms/cassandra/sink_cluster/mod.rs
index 6a5ced7a6..75ab1975d 100644
--- a/shotover-proxy/src/transforms/cassandra/sink_cluster/mod.rs
+++ b/shotover-proxy/src/transforms/cassandra/sink_cluster/mod.rs
@@ -223,15 +223,10 @@ impl CassandraSinkCluster {
                     .iter()
                     .any(|x| x.address == address && x.is_up)
                 {
-                    let addresses: Vec<_> = self
-                        .pool
-                        .get_shuffled_nodes_in_dc_rack(
-                            &self.local_shotover_node.rack,
-                            &mut self.rng,
-                        )
-                        .iter()
-                        .map(|node| node.address)
-                        .collect();
+                    let addresses = self.pool.get_shuffled_nodes_in_dc_rack(
+                        &self.local_shotover_node.rack,
+                        &mut self.rng,
+                    );
                     self.create_control_connection(&addresses).await.map_err(|e|
                         e.context("Failed to recreate control connection after control connection node went down")
                     )?;
@@ -272,9 +267,6 @@ impl CassandraSinkCluster {
             } else {
                 self.pool
                     .get_shuffled_nodes_in_dc_rack(&self.local_shotover_node.rack, &mut self.rng)
-                    .iter()
-                    .map(|node| node.address)
-                    .collect()
             };
 
             self.create_control_connection(&points)
@@ -526,12 +518,9 @@ impl CassandraSinkCluster {
             // If we have to populate the local_nodes at this point then that means the control connection
             // may not have been made against a node in the configured data_center/rack.
             // Therefore we need to recreate the control connection to ensure that it is in the configured data_center/rack.
-            let addresses: Vec<_> = self
+            let addresses = self
                 .pool
-                .get_shuffled_nodes_in_dc_rack(&self.local_shotover_node.rack, &mut self.rng)
-                .iter()
-                .map(|node| node.address)
-                .collect();
+                .get_shuffled_nodes_in_dc_rack(&self.local_shotover_node.rack, &mut self.rng);
             self.create_control_connection(&addresses)
                 .await
                 .map_err(|e| e.context("Failed to recreate control connection when initial connection was possibly against the wrong node"))?;
diff --git a/shotover-proxy/src/transforms/cassandra/sink_cluster/node_pool.rs b/shotover-proxy/src/transforms/cassandra/sink_cluster/node_pool.rs
index a987dfa95..ba49e225a 100644
--- a/shotover-proxy/src/transforms/cassandra/sink_cluster/node_pool.rs
+++ b/shotover-proxy/src/transforms/cassandra/sink_cluster/node_pool.rs
@@ -8,8 +8,8 @@ use cassandra_protocol::frame::Version;
 use cassandra_protocol::token::Murmur3Token;
 use cassandra_protocol::types::CBytesShort;
 use rand::prelude::*;
-use std::collections::HashMap;
 use std::sync::Arc;
+use std::{collections::HashMap, net::SocketAddr};
 use tokio::sync::{watch, RwLock};
 
 pub enum GetReplicaErr {
@@ -77,11 +77,12 @@ impl NodePool {
         &mut self,
         rack: &str,
         rng: &mut SmallRng,
-    ) -> Vec<&mut CassandraNode> {
+    ) -> Vec<SocketAddr> {
         let mut nodes: Vec<_> = self
             .nodes
             .iter_mut()
             .filter(|node| node.is_up && node.rack == *rack)
+            .map(|node| node.address)
             .collect();
 
         nodes.shuffle(rng);