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));