diff --git a/include/mbedtls/ssl_internal.h b/include/mbedtls/ssl_internal.h
index bd2169e..5d8f6fd 100644
--- a/include/mbedtls/ssl_internal.h
+++ b/include/mbedtls/ssl_internal.h
@@ -573,10 +573,10 @@
 
 #if defined(MBEDTLS_USE_TINYCRYPT)
     uint8_t ecdh_privkey[NUM_ECC_BYTES];
-#if defined(MBEDTLS_EARLY_KEY_COMPUTATION)
+#if defined(MBEDTLS_SSL_EARLY_KEY_COMPUTATION)
     uint8_t ecdhe_computed;
     uint8_t ecdh_publickey[2*NUM_ECC_BYTES];
-#endif /* MBEDTLS_EARLY_KEY_COMPUTATION */
+#endif /* MBEDTLS_SSL_EARLY_KEY_COMPUTATION */
     uint8_t ecdh_peerkey[2*NUM_ECC_BYTES];
 #endif /* MBEDTLS_USE_TINYCRYPT */