From d681d1a21904697b42e640fa46c242a444db7734 Mon Sep 17 00:00:00 2001
From: Revertron <r@revertron.com>
Date: Mon, 19 Apr 2021 15:44:54 +0200
Subject: [PATCH] Fixed build.

---
 build.rs           | 4 ++--
 src/p2p/network.rs | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/build.rs b/build.rs
index 51d8092..eb36cb7 100644
--- a/build.rs
+++ b/build.rs
@@ -48,14 +48,14 @@ fn download_iana_zones(zones_name: &str, hashes_name: &str) {
         Ok(mut file) => {
             file.write_all(zones.trim().as_bytes()).expect("Error saving TLDs file!");
         }
-        Err(e) => { println!("Error opening TLDs file!\n{}", e); }
+        Err(e) => { panic!("Error opening TLDs file!\n{}", e); }
     }
 
     match File::create(Path::new(hashes_name)) {
         Ok(mut file) => {
             file.write_all(hashes.trim().as_bytes()).expect("Error saving TLD-hashes file!");
         }
-        Err(e) => { println!("Error opening TLD-hashes file!\n{}", e); }
+        Err(e) => { panic!("Error opening TLD-hashes file!\n{}", e); }
     }
 }
 
diff --git a/src/p2p/network.rs b/src/p2p/network.rs
index 6060456..98eef91 100644
--- a/src/p2p/network.rs
+++ b/src/p2p/network.rs
@@ -477,6 +477,7 @@ fn handle_message(context: Arc<Mutex<Context>>, message: Message, peers: &mut Pe
 }
 
 fn process_new_block(context: Arc<Mutex<Context>>, peers: &mut Peers, token: &Token, block: Block) -> State {
+    let peers_count = peers.get_peers_active_count();
     let peer = peers.get_mut_peer(token).unwrap();
     peer.set_received_block(block.index);
     if let Some(transaction) = &block.transaction {
@@ -485,7 +486,7 @@ fn process_new_block(context: Arc<Mutex<Context>>, peers: &mut Peers, token: &To
             return State::Banned;
         }
     }
-    let peers_count = peers.get_peers_active_count();
+
     let mut context = context.lock().unwrap();
     let max_height = context.chain.max_height();
     match context.chain.check_new_block(&block) {