diff --git a/programs/pkey/dh_client.c b/programs/pkey/dh_client.c
index 9dd38bc..0d4172b 100644
--- a/programs/pkey/dh_client.c
+++ b/programs/pkey/dh_client.c
@@ -57,10 +57,10 @@
     mbedtls_exit(0);
 }
 
-#elif defined(MBEDTLS_CIPHER_ENCRYPT_ONLY)
+#elif defined(MBEDTLS_BLOCK_CIPHER_NO_DECRYPT)
 int main(void)
 {
-    mbedtls_printf("MBEDTLS_CIPHER_ENCRYPT_ONLY implicitly defined.\n");
+    mbedtls_printf("MBEDTLS_BLOCK_CIPHER_NO_DECRYPT implicitly defined.\n");
     mbedtls_exit(0);
 }
 #else