From b652ddb99f9124dcdb316aa10ec29054362a131b Mon Sep 17 00:00:00 2001
From: chriseth <chris@ethereum.org>
Date: Tue, 13 Sep 2022 14:23:47 +0200
Subject: [PATCH] Fix.

---
 src/flow/flow.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/flow/flow.rs b/src/flow/flow.rs
index cd71773..d66bdf7 100644
--- a/src/flow/flow.rs
+++ b/src/flow/flow.rs
@@ -115,12 +115,13 @@ fn extract_transfers(
             account_balances.len()
         );
         let edge = next_full_capacity_edge(&mut used_edges, &mut account_balances);
+        assert!(account_balances.contains_key(&edge.from));
         account_balances
             .entry(edge.from)
             .and_modify(|balance| *balance -= edge.capacity);
-        account_balances
+        *account_balances
             .entry(edge.to)
-            .and_modify(|balance| *balance += edge.capacity);
+            .or_default() += edge.capacity;
         account_balances.retain(|_account, balance| balance > &mut U256::from(0));
         used_edges
             .entry(Node::Node(edge.from))