diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index dbfcde2..e3ce590 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -4216,7 +4216,7 @@
             return( MBEDTLS_ERR_SSL_FEATURE_UNAVAILABLE );
         }
     }
-#endif
+#endif /* MBEDTLS_SSL_PROTO_TLS */
 
     return( 0 );
 }
@@ -10847,7 +10847,7 @@
         *major = ver[0];
         *minor = ver[1];
     }
-#endif
+#endif /* MBEDTLS_SSL_PROTO_TLS */
 }
 
 int mbedtls_ssl_set_calc_verify_md( mbedtls_ssl_context *ssl, int md )