diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index d0f1ccc..4a111c6 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -3749,7 +3749,7 @@
     ssl->alpn_chosen = NULL;
 #endif
 
-#if defined(POLARSSL_SSL_PROTO_DTLS) && defined(POLARSSL_SSL_SRV_C)
+#if defined(POLARSSL_SSL_DTLS_HELLO_VERIFY)
     polarssl_free( ssl->cli_id );
     ssl->cli_id = NULL;
     ssl->cli_id_len = 0;
@@ -5039,7 +5039,7 @@
     }
 #endif
 
-#if defined(POLARSSL_SSL_PROTO_DTLS) && defined(POLARSSL_SSL_SRV_C)
+#if defined(POLARSSL_SSL_DTLS_HELLO_VERIFY)
     polarssl_free( ssl->cli_id );
     md_free( &ssl->hvr_hmac_ctx );
 #endif