diff --git a/quic/test_tools/quic_test_utils.cc b/quic/test_tools/quic_test_utils.cc
index 680b16c..bfbf512 100644
--- a/quic/test_tools/quic_test_utils.cc
+++ b/quic/test_tools/quic_test_utils.cc
@@ -582,7 +582,7 @@
     size_t write_length,
     QuicStreamOffset offset,
     StreamSendingState state,
-    bool /*is_retransmission*/,
+    TransmissionType /*type*/,
     quiche::QuicheOptional<EncryptionLevel> /*level*/) {
   if (write_length > 0) {
     auto buf = std::make_unique<char[]>(write_length);
@@ -656,7 +656,7 @@
     size_t write_length,
     QuicStreamOffset offset,
     StreamSendingState state,
-    bool /*is_retransmission*/,
+    TransmissionType /*type*/,
     quiche::QuicheOptional<EncryptionLevel> /*level*/) {
   if (write_length > 0) {
     auto buf = std::make_unique<char[]>(write_length);