From 0a7e199c823c283c7607e72a080ebe452dc03b65 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Fri, 4 Jan 2019 13:25:47 -0800
Subject: [PATCH] Don't follow the leader: assume drone runs on the network
 entrypoint

---
 wallet/src/wallet.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/wallet/src/wallet.rs b/wallet/src/wallet.rs
index 42b4c6f07b..935aae6f7b 100644
--- a/wallet/src/wallet.rs
+++ b/wallet/src/wallet.rs
@@ -101,8 +101,9 @@ impl Default for WalletConfig {
 }
 
 impl WalletConfig {
-    pub fn drone_addr(&self, tpu_addr: SocketAddr) -> SocketAddr {
-        let mut drone_addr = tpu_addr;
+    pub fn drone_addr(&self) -> SocketAddr {
+        // Assume drone is running on the provided network entrypoint
+        let mut drone_addr = self.network;
         drone_addr.set_port(self.drone_port.unwrap_or(DRONE_PORT));
         drone_addr
     }
@@ -311,8 +312,7 @@ pub fn process_command(config: &WalletConfig) -> Result<String, Box<dyn error::E
     }
 
     let leader = poll_gossip_for_leader(config.network, config.timeout)?;
-    let tpu_addr = leader.tpu;
-    let drone_addr = config.drone_addr(tpu_addr);
+    let drone_addr = config.drone_addr();
     let rpc_addr = config.rpc_addr(leader.rpc);
     let rpc_client = RpcClient::new(rpc_addr);