diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 4615920..a8ccde5 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -6419,13 +6419,14 @@
         return PSA_ERROR_BAD_STATE;
     }
 
-    if (data > 0xFFFFFFFF) {
+    if (data > PSA_VENDOR_PBKDF2_MAX_ITERATIONS) {
         return PSA_ERROR_NOT_SUPPORTED;
     }
 
     if (data == 0) {
         return PSA_ERROR_INVALID_ARGUMENT;
     }
+
     pbkdf2->input_cost = data;
     pbkdf2->state = PSA_PBKDF2_STATE_INPUT_COST_SET;