X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Fln%2Freload_tests.rs;h=b3d52b78f2b5a5a3921c66873325fde418601b2b;hb=90f24a6a509157ad8f5027dab1f287b9a6acca08;hp=223aa5dbac30ba64604992d6b272722ceb3ed1c0;hpb=ec1a66e4cd241d4c8d3ff8cf512d0e5108a0a710;p=rust-lightning

diff --git a/lightning/src/ln/reload_tests.rs b/lightning/src/ln/reload_tests.rs
index 223aa5db..b3d52b78 100644
--- a/lightning/src/ln/reload_tests.rs
+++ b/lightning/src/ln/reload_tests.rs
@@ -1065,9 +1065,10 @@ fn do_forwarded_payment_no_manager_persistence(use_cs_commitment: bool, claim_ht
 			confirm_transaction(&nodes[1], &cs_commitment_tx[1]);
 		} else {
 			connect_blocks(&nodes[1], htlc_expiry - nodes[1].best_block_info().1 + 1);
-			let bs_htlc_timeout_tx = nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap().split_off(0);
-			assert_eq!(bs_htlc_timeout_tx.len(), 1);
-			confirm_transaction(&nodes[1], &bs_htlc_timeout_tx[0]);
+			let mut txn = nodes[1].tx_broadcaster.txn_broadcast();
+			assert_eq!(txn.len(), if nodes[1].connect_style.borrow().updates_best_block_first() { 2 } else { 1 });
+			let bs_htlc_timeout_tx = txn.pop().unwrap();
+			confirm_transaction(&nodes[1], &bs_htlc_timeout_tx);
 		}
 	} else {
 		confirm_transaction(&nodes[1], &bs_commitment_tx[0]);