diff --git a/quic/core/quic_connection_test.cc b/quic/core/quic_connection_test.cc
index 56b0453..5d35353 100644
--- a/quic/core/quic_connection_test.cc
+++ b/quic/core/quic_connection_test.cc
@@ -806,8 +806,7 @@
   }
 
   QuicByteCount GetBytesInFlight() {
-    return QuicSentPacketManagerPeer::GetBytesInFlight(
-        QuicConnectionPeer::GetSentPacketManager(this));
+    return QuicConnectionPeer::GetSentPacketManager(this)->GetBytesInFlight();
   }
 
   void set_notifier(SimpleSessionNotifier* notifier) { notifier_ = notifier; }
@@ -3375,12 +3374,12 @@
 
   // Ensure that the data is still in flight, but the retransmission alarm is no
   // longer set.
-  EXPECT_GT(QuicSentPacketManagerPeer::GetBytesInFlight(manager_), 0u);
+  EXPECT_GT(manager_->GetBytesInFlight(), 0u);
   if (GetQuicReloadableFlag(quic_optimize_inflight_check)) {
     EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet());
     // Firing the alarm should remove all bytes_in_flight.
     connection_.GetRetransmissionAlarm()->Fire();
-    EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_));
+    EXPECT_EQ(0u, manager_->GetBytesInFlight());
   }
   EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet());
 }
@@ -3654,7 +3653,7 @@
     EXPECT_TRUE(connection_.GetRetransmissionAlarm()->IsSet());
     // Firing the alarm should remove all bytes_in_flight.
     connection_.GetRetransmissionAlarm()->Fire();
-    EXPECT_EQ(0u, QuicSentPacketManagerPeer::GetBytesInFlight(manager_));
+    EXPECT_EQ(0u, manager_->GetBytesInFlight());
   }
   EXPECT_FALSE(connection_.GetRetransmissionAlarm()->IsSet());
   EXPECT_FALSE(QuicConnectionPeer::HasRetransmittableFrames(&connection_, 2));