diff --git a/crates/dig-node-core/src/seams/chia_peer/coinset_resolver.rs b/crates/dig-node-core/src/seams/chia_peer/coinset_resolver.rs index 7aadd0a5..01d7c1f2 100644 --- a/crates/dig-node-core/src/seams/chia_peer/coinset_resolver.rs +++ b/crates/dig-node-core/src/seams/chia_peer/coinset_resolver.rs @@ -415,7 +415,7 @@ mod tests { async fn unspent_coins_by_hint(&self, _hint: ChiaBytes32) -> ChainResult> { match self .answers_left - .fetch_update(Ordering::SeqCst, Ordering::SeqCst, |left| { + .try_update(Ordering::SeqCst, Ordering::SeqCst, |left| { left.checked_sub(1) }) { Ok(_) => Ok(Vec::new()), diff --git a/crates/dig-node-core/src/seams/dig_peer/module_reshare.rs b/crates/dig-node-core/src/seams/dig_peer/module_reshare.rs index aebc7808..675658c5 100644 --- a/crates/dig-node-core/src/seams/dig_peer/module_reshare.rs +++ b/crates/dig-node-core/src/seams/dig_peer/module_reshare.rs @@ -1622,7 +1622,7 @@ mod tests { ) -> Result, dig_download::DownloadError> { let spend = self .budget - .fetch_update(Ordering::SeqCst, Ordering::SeqCst, |left| { + .try_update(Ordering::SeqCst, Ordering::SeqCst, |left| { (left > 0).then(|| left.saturating_sub(1)) }); if spend.is_err() {