diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index bcf44b7..0597298 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -3050,7 +3050,7 @@
                 while ((ret = mbedtls_ssl_write_early_data(&ssl, buf + written,
                                                            len - written)) < 0) {
                     if (ret == MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA) {
-                        break;
+                        goto end_of_early_data;
                     }
                     if (ret != MBEDTLS_ERR_SSL_WANT_READ &&
                         ret != MBEDTLS_ERR_SSL_WANT_WRITE &&
@@ -3069,15 +3069,14 @@
 #endif
                     }
                 }
-                if (ret == MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA) {
-                    break;
-                }
 
                 frags++;
                 written += ret;
             } while (written < len);
         }
 
+end_of_early_data:
+
         buf[written] = '\0';
         mbedtls_printf(
             " %" MBEDTLS_PRINTF_SIZET " bytes of early data written in %" MBEDTLS_PRINTF_SIZET " fragments\n\n%s\n",