diff --git a/quiche/quic/core/quic_connection_id_manager_test.cc b/quiche/quic/core/quic_connection_id_manager_test.cc
index c0f1333..bb85d5d 100644
--- a/quiche/quic/core/quic_connection_id_manager_test.cc
+++ b/quiche/quic/core/quic_connection_id_manager_test.cc
@@ -548,7 +548,8 @@
   QuicConnectionId CheckGenerate(QuicConnectionId old_cid) {
     QuicConnectionId new_cid =
         QuicUtils::CreateReplacementConnectionId(old_cid);
-    if (GetQuicRestartFlag(quic_abstract_connection_id_generator)) {
+    if (GetQuicReloadableFlag(
+            quic_connection_uses_abstract_connection_id_generator)) {
       // Ready for the actual call.
       EXPECT_CALL(connection_id_generator_, GenerateNextConnectionId(old_cid))
           .WillOnce(Return(new_cid));