diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index 1d44ccf..c46f041 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -4949,7 +4949,7 @@
 #if defined(MBEDTLS_ECP_DP_BP512R1_ENABLED)
     MBEDTLS_SSL_IANA_TLS_GROUP_BP512R1,
 #endif
-#if defined(MBEDTLS_DHM_C)
+#if defined(PSA_WANT_ALG_FFDH)
     MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE2048,
     MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE3072,
     MBEDTLS_SSL_IANA_TLS_GROUP_FFDHE4096,