diff --git a/tests/src/psa_exercise_key.c b/tests/src/psa_exercise_key.c
index 53af7b8..9576afd 100644
--- a/tests/src/psa_exercise_key.c
+++ b/tests/src/psa_exercise_key.c
@@ -164,7 +164,7 @@
                                 psa_algorithm_t alg )
 {
     psa_cipher_operation_t operation = PSA_CIPHER_OPERATION_INIT;
-    unsigned char iv[16] = {0};
+    unsigned char iv[PSA_CIPHER_IV_MAX_SIZE] = {0};
     size_t iv_length;
     psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
     psa_key_type_t key_type;
@@ -242,7 +242,7 @@
                               psa_key_usage_t usage,
                               psa_algorithm_t alg )
 {
-    unsigned char nonce[16] = {0};
+    unsigned char nonce[PSA_AEAD_NONCE_MAX_SIZE] = {0};
     size_t nonce_length;
     psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
     psa_key_type_t key_type;