diff --git a/quic/core/http/end_to_end_test.cc b/quic/core/http/end_to_end_test.cc
index 5d2d884..cb05e78 100644
--- a/quic/core/http/end_to_end_test.cc
+++ b/quic/core/http/end_to_end_test.cc
@@ -1651,7 +1651,7 @@
 
   // The 0-RTT handshake should succeed.
   client_->Connect();
-  EXPECT_TRUE(client_->client()->WaitForOneRttKeysAvailable());
+  EXPECT_TRUE(client_->client()->WaitForHandshakeConfirmed());
   ASSERT_TRUE(client_->client()->connected());
   SendSynchronousFooRequestAndCheckResponse();
 
@@ -1722,7 +1722,7 @@
 
   // Client re-connect.
   client_->Connect();
-  ASSERT_TRUE(client_->client()->WaitForOneRttKeysAvailable());
+  ASSERT_TRUE(client_->client()->WaitForHandshakeConfirmed());
   client_->WaitForWriteToFlush();
   client_->WaitForResponse();
   ASSERT_TRUE(client_->client()->connected());