diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index 867dd4a..f7ed759 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -2654,7 +2654,7 @@
             *key_size = 256;
             break;
 #endif
-#if defined(MBEDTLS_SSL_HAVE_CHACHAPOLY)
+#if defined(PSA_WANT_ALG_CHACHA20_POLY1305)
         case MBEDTLS_CIPHER_CHACHA20_POLY1305:
             *alg = PSA_ALG_CHACHA20_POLY1305;
             *key_type = PSA_KEY_TYPE_CHACHA20;