diff --git a/quic/core/quic_connection.cc b/quic/core/quic_connection.cc
index 1e59d4f..cdb93a3 100644
--- a/quic/core/quic_connection.cc
+++ b/quic/core/quic_connection.cc
@@ -359,6 +359,9 @@
   if (fix_missing_connected_checks_) {
     QUIC_RELOADABLE_FLAG_COUNT(quic_add_missing_connected_checks);
   }
+  if (enable_aead_limits_) {
+    QUIC_RELOADABLE_FLAG_COUNT(quic_enable_aead_limits);
+  }
   QUIC_DLOG(INFO) << ENDPOINT << "Created connection with server connection ID "
                   << server_connection_id
                   << " and version: " << ParsedQuicVersionToString(version());
diff --git a/quic/core/quic_framer.cc b/quic/core/quic_framer.cc
index 03b27f2..39be55e 100644
--- a/quic/core/quic_framer.cc
+++ b/quic/core/quic_framer.cc
@@ -4247,6 +4247,9 @@
 void QuicFramer::SetKeyUpdateSupportForConnection(bool enabled) {
   QUIC_DVLOG(1) << ENDPOINT << "SetKeyUpdateSupportForConnection: " << enabled;
   support_key_update_for_connection_ = enabled;
+  if (enabled) {
+    QUIC_RELOADABLE_FLAG_COUNT(quic_key_update_supported);
+  }
 }
 
 void QuicFramer::DiscardPreviousOneRttKeys() {