diff --git a/src/kex.c b/src/kex.c
index 1101f876..7a1c89c7 100644
--- a/src/kex.c
+++ b/src/kex.c
@@ -745,7 +745,7 @@ int ssh_set_client_kex(ssh_session session)
     }
 
     /* For rekeying, skip the extension negotiation */
-    if (session->session_state == SSH_SESSION_STATE_AUTHENTICATED) {
+    if (session->flags & SSH_SESSION_FLAG_AUTHENTICATED) {
         return SSH_OK;
     }