X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=src%2Flib.rs;fp=src%2Flib.rs;h=25ce1a2f5015c2a23dd76479fd12f831a854903d;hb=694b11978177600415c2183f1eeac73471b07f62;hp=cc5a350847e2c4c42322d2bf20545d5de900059a;hpb=32ac36a7f09ee15909420e65edf387c029651af1;p=rapid-gossip-sync-server

diff --git a/src/lib.rs b/src/lib.rs
index cc5a350..25ce1a2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -16,7 +16,6 @@ use std::io::BufReader;
 use std::sync::Arc;
 
 use bitcoin::blockdata::constants::genesis_block;
-use bitcoin::Network;
 use bitcoin::secp256k1::PublicKey;
 use lightning::routing::gossip::NetworkGraph;
 use lightning::util::ser::{ReadableArgs, Writeable};
@@ -54,6 +53,7 @@ pub struct SerializedResponse {
 
 impl RapidSyncProcessor {
 	pub fn new() -> Self {
+		let network = config::network();
 		let logger = TestLogger::new();
 		let network_graph = if let Ok(file) = File::open(&config::network_graph_cache_path()) {
 			println!("Initializing from cached network graph…");
@@ -65,10 +65,10 @@ impl RapidSyncProcessor {
 				network_graph
 			} else {
 				println!("Initialization from cached network graph failed: {}", network_graph_result.err().unwrap());
-				NetworkGraph::new(genesis_block(Network::Bitcoin).header.block_hash(), logger)
+				NetworkGraph::new(genesis_block(network).header.block_hash(), logger)
 			}
 		} else {
-			NetworkGraph::new(genesis_block(Network::Bitcoin).header.block_hash(), logger)
+			NetworkGraph::new(genesis_block(network).header.block_hash(), logger)
 		};
 		let arc_network_graph = Arc::new(network_graph);
 		Self {