diff --git a/library/cipher_wrap.h b/library/cipher_wrap.h
index 2cbc216..bf1291d 100644
--- a/library/cipher_wrap.h
+++ b/library/cipher_wrap.h
@@ -93,7 +93,7 @@
     int (*setkey_enc_func)(void *ctx, const unsigned char *key,
                            unsigned int key_bitlen);
 
-#if !defined(MBEDTLS_CIPHER_ENCRYPT_ONLY)
+#if !defined(MBEDTLS_BLOCK_CIPHER_NO_DECRYPT)
     /** Set key for decryption purposes */
     int (*setkey_dec_func)(void *ctx, const unsigned char *key,
                            unsigned int key_bitlen);