diff --git a/pq/tls/client-pq-tls13.c b/pq/tls/client-pq-tls13.c
index 22aeb67c..3d863f90 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_HAVE_MLKEM) && \
     defined(HAVE_DILITHIUM)
 
 #define DEFAULT_PORT 11111
diff --git a/pq/tls/server-pq-tls13.c b/pq/tls/server-pq-tls13.c
index 0915f978..b23504a6 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_HAVE_MLKEM) && \
     defined(HAVE_DILITHIUM)
 
 #define DEFAULT_PORT 11111
@@ -248,13 +248,6 @@ int main(int argc, char** argv)
             ret = -1; goto exit;
         }
 
-        ret = wolfSSL_UseKeyShare(ssl, WOLFSSL_P521_ML_KEM_1024);
-        if (ret < 0) {
-            fprintf(stderr, "ERROR: failed to set the requested group to "
-                            "WOLFSSL_P521_ML_KEM_1024.\n");
-            ret = -1; goto exit;
-        }
-
         /* Attach wolfSSL to the socket */
         wolfSSL_set_fd(ssl, mConnd);