diff --git a/library/ssl_msg.c b/library/ssl_msg.c
index c61a7bf..52bd826 100644
--- a/library/ssl_msg.c
+++ b/library/ssl_msg.c
@@ -6130,7 +6130,7 @@
         }
 
         remaining = ssl->session_negotiate->max_early_data_size -
-                    ssl->early_data_count;
+                    ssl->total_early_data_size;
 
         if (remaining == 0) {
             return MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA;
@@ -6154,7 +6154,7 @@
     }
 
     written_data_len = ssl_write_real(ssl, buf, len);
-    ssl->early_data_count += written_data_len;
+    ssl->total_early_data_size += written_data_len;
 
     MBEDTLS_SSL_DEBUG_MSG(2, ("<= write early_data, len=%d", written_data_len));
 
diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index ee72179..5bedd91 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -1105,7 +1105,7 @@
 #if defined(MBEDTLS_SSL_SRV_C)
     ssl->discard_early_data_record = MBEDTLS_SSL_EARLY_DATA_NO_DISCARD;
 #endif
-    ssl->early_data_count = 0;
+    ssl->total_early_data_size = 0;
 #endif /* MBEDTLS_SSL_EARLY_DATA */
 
     /* Initialize structures */