diff --git a/pq/tls/client-pq-tls13.c b/pq/tls/client-pq-tls13.c
index 9a708308..4cd52e20 100644
--- a/pq/tls/client-pq-tls13.c
+++ b/pq/tls/client-pq-tls13.c
@@ -36,7 +36,7 @@
 #include <wolfssl/wolfio.h>
 #include <wolfssl/wolfcrypt/error-crypt.h>
 
-#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_HAVE_KYBER) && \
+#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_WC_MLKEM) && \
     defined(HAVE_DILITHIUM)
 
 #define DEFAULT_PORT 11111
@@ -116,7 +116,7 @@ static int Tls13SecretCallback(WOLFSSL* ssl, int id, const unsigned char* secret
 int main(int argc, char** argv)
 {
     int ret = 0;
-#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_HAVE_KYBER) && \
+#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_WC_MLKEM) && \
     defined(HAVE_DILITHIUM)
     int                sockfd = SOCKET_INVALID;
     struct sockaddr_in servAddr;
diff --git a/pq/tls/server-pq-tls13.c b/pq/tls/server-pq-tls13.c
index 95213f81..556157be 100644
--- a/pq/tls/server-pq-tls13.c
+++ b/pq/tls/server-pq-tls13.c
@@ -41,7 +41,7 @@
 #include <wolfssl/wolfio.h>
 #include <wolfssl/wolfcrypt/error-crypt.h>
 
-#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_HAVE_KYBER) && \
+#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_WC_MLKEM) && \
     defined(HAVE_DILITHIUM)
 
 #define DEFAULT_PORT 11111
@@ -145,7 +145,7 @@ int main(int argc, char** argv)
 {
     int ret = 0;
 
-#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_HAVE_KYBER) && \
+#if defined(WOLFSSL_TLS13) && defined(WOLFSSL_WC_MLKEM) && \
     defined(HAVE_DILITHIUM)
 
     struct sockaddr_in servAddr;