Don't count NotYetReceived packet as lost

1 files changed, 3 insertions, 3 deletions

diff --git a/src/stats.rs b/src/stats.rs
index 3848d6b..afbed9a 100644
--- a/src/stats.rs
+++ b/src/stats.rs
@@ -17,7 +17,7 @@ pub fn compute_stats(states: &[crate::state::State]) -> Option<Stats> {
 
     let received = states
         .iter()
-        .filter(|state| matches!(state, crate::state::State::Received { .. }))
+        .filter(|state| matches!(state, crate::state::State::Received { .. } | crate::state::State::NotYetReceived  { .. } ))
         .count();
     let lost = states.len() - received;
 
@@ -32,8 +32,8 @@ pub fn compute_stats(states: &[crate::state::State]) -> Option<Stats> {
     }
 
     // RTT stats
-    let min_rtt = rtts.iter().min().copied().unwrap();
-    let max_rtt = rtts.iter().max().copied().unwrap();
+    let min_rtt = rtts.iter().min().copied().unwrap_or(0);
+    let max_rtt = rtts.iter().max().copied().unwrap_or(0);
 
     let sum_rtt: u64 = rtts.iter().sum();
     let avg_rtt = sum_rtt as f64 / rtts.len() as f64;