diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 59169d5..af4ab65 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -6423,11 +6423,11 @@
     if (pbkdf2->state != PSA_PBKDF2_STATE_INIT) {
         return PSA_ERROR_BAD_STATE;
     }
-#if UINT_MAX > 0xFFFFFFFF
+
     if (data > 0xFFFFFFFF) {
-        return PSA_ERROR_INVALID_ARGUMENT;
+        return PSA_ERROR_NOT_SUPPORTED;
     }
-#endif
+
     if (data == 0) {
         return PSA_ERROR_INVALID_ARGUMENT;
     }