diff --git a/library/ssl_msg.c b/library/ssl_msg.c
index ccced0a..56e5514 100644
--- a/library/ssl_msg.c
+++ b/library/ssl_msg.c
@@ -6098,7 +6098,7 @@
 
     /*
      * If we are at the beginning of the handshake, the early data state being
-     * equal to MBEDTLS_SSL_EARLY_DATA_STATE_UNKNOWN or
+     * equal to MBEDTLS_SSL_EARLY_DATA_STATE_IDLE or
      * MBEDTLS_SSL_EARLY_DATA_STATE_SENT advance the handshake just
      * enough to be able to send early data if possible. That way, we can
      * guarantee that when starting the handshake with this function we will
@@ -6108,9 +6108,9 @@
      * as the early data outbound transform has not been set as we may have to
      * first send a dummy CCS in clear.
      */
-    if ((ssl->early_data_state == MBEDTLS_SSL_EARLY_DATA_STATE_UNKNOWN) ||
+    if ((ssl->early_data_state == MBEDTLS_SSL_EARLY_DATA_STATE_IDLE) ||
         (ssl->early_data_state == MBEDTLS_SSL_EARLY_DATA_STATE_SENT)) {
-        while ((ssl->early_data_state == MBEDTLS_SSL_EARLY_DATA_STATE_UNKNOWN) ||
+        while ((ssl->early_data_state == MBEDTLS_SSL_EARLY_DATA_STATE_IDLE) ||
                (ssl->early_data_state == MBEDTLS_SSL_EARLY_DATA_STATE_SENT)) {
             ret = mbedtls_ssl_handshake_step(ssl);
             if (ret != 0) {