diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 57533cc492..3ec92cc061 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -7488,7 +7488,7 @@ static psa_status_t psa_key_derivation_input_internal(
     psa_status_t status;
     psa_algorithm_t kdf_alg = psa_key_derivation_get_kdf_alg(operation);
 
-    if (kdf_alg == 0) {
+    if (kdf_alg == PSA_ALG_NONE) {
         /* This is a blank or aborted operation. */
         status = PSA_ERROR_BAD_STATE;
         goto exit;
@@ -7552,7 +7552,7 @@ static psa_status_t psa_key_derivation_input_integer_internal(
     psa_status_t status;
     psa_algorithm_t kdf_alg = psa_key_derivation_get_kdf_alg(operation);
 
-    if (kdf_alg == 0) {
+    if (kdf_alg == PSA_ALG_NONE) {
         /* This is a blank or aborted operation. */
         status = PSA_ERROR_BAD_STATE;
         goto exit;