diff --git a/include/mbedtls/config.h b/include/mbedtls/config.h
index 3c3c1ce..0fca50a 100644
--- a/include/mbedtls/config.h
+++ b/include/mbedtls/config.h
@@ -1616,13 +1616,13 @@
 
 
 /**
- * \def MBEDTLS_IMMEDIATE_TRANSMISSION
+ * \def MBEDTLS_SSL_IMMEDIATE_TRANSMISSION
  *
  * Force stack to immediately transmit messages.
  *
  * Requires: MBEDTLS_SSL_PROTO_DTLS
  */
-//#define MBEDTLS_IMMEDIATE_TRANSMISSION
+//#define MBEDTLS_SSL_IMMEDIATE_TRANSMISSION
 
 /**
  * \def MBEDTLS_EARLY_KEY_COMPUTATION
diff --git a/include/mbedtls/ssl_internal.h b/include/mbedtls/ssl_internal.h
index 80da3ac..bd2169e 100644
--- a/include/mbedtls/ssl_internal.h
+++ b/include/mbedtls/ssl_internal.h
@@ -1203,7 +1203,7 @@
 void mbedtls_ssl_recv_flight_completed( mbedtls_ssl_context *ssl );
 int mbedtls_ssl_resend( mbedtls_ssl_context *ssl );
 int mbedtls_ssl_flight_transmit( mbedtls_ssl_context *ssl );
-#if defined(MBEDTLS_IMMEDIATE_TRANSMISSION)
+#if defined(MBEDTLS_SSL_IMMEDIATE_TRANSMISSION)
 void mbedtls_ssl_immediate_flight_done( mbedtls_ssl_context *ssl );
 #endif
 #endif