diff --git a/library/pk_wrap.h b/library/pk_wrap.h
index d4a90ab..e310005 100644
--- a/library/pk_wrap.h
+++ b/library/pk_wrap.h
@@ -155,6 +155,6 @@
                              unsigned char *sig, size_t sig_size,
                              size_t *sig_len );
 
-#endif /* MBEDTLS_USE_PSA_CRYPTO */
+#endif /* MBEDTLS_PSA_CRYPTO_C */
 
 #endif /* MBEDTLS_PK_WRAP_H */