diff --git a/library/ssl_misc.h b/library/ssl_misc.h
index b1ae119..2235497 100644
--- a/library/ssl_misc.h
+++ b/library/ssl_misc.h
@@ -255,7 +255,7 @@
 
 /* This macro determines whether CBC is supported. */
 #if defined(MBEDTLS_SSL_HAVE_CBC)      &&                                  \
-    (defined(MBEDTLS_SSL_HAVE_AES)     ||                                  \
+    (defined(PSA_WANT_KEY_TYPE_AES)     ||                                  \
     defined(MBEDTLS_SSL_HAVE_CAMELLIA) ||                                  \
     defined(MBEDTLS_SSL_HAVE_ARIA))
 #define MBEDTLS_SSL_SOME_SUITES_USE_CBC