diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 95f9740..e40e370 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -3749,11 +3749,8 @@
     return( status );
 }
 
-static psa_status_t psa_aead_final_checks( psa_aead_operation_t *operation,
-                                           size_t output_size )
+static psa_status_t psa_aead_final_checks( psa_aead_operation_t *operation )
 {
-    size_t finish_output_size;
-
     if( operation->id == 0 || !operation->nonce_set )
         return( PSA_ERROR_BAD_STATE );
 
@@ -3761,13 +3758,6 @@
                                    operation->body_remaining != 0 ) )
         return( PSA_ERROR_INVALID_ARGUMENT );
 
-    finish_output_size = operation->is_encrypt ?
-         PSA_AEAD_FINISH_OUTPUT_SIZE( operation->key_type, operation->alg ) :
-         PSA_AEAD_VERIFY_OUTPUT_SIZE( operation->key_type, operation->alg );
-
-    if( output_size < finish_output_size )
-        return( PSA_ERROR_BUFFER_TOO_SMALL );
-
     return( PSA_SUCCESS );
 }
 
@@ -3785,7 +3775,7 @@
     *ciphertext_length = 0;
     *tag_length = tag_size;
 
-    status = psa_aead_final_checks( operation, ciphertext_size );
+    status = psa_aead_final_checks( operation );
 
     if( status != PSA_SUCCESS )
         goto exit;
@@ -3831,7 +3821,7 @@
 
     *plaintext_length = 0;
 
-    status = psa_aead_final_checks( operation, plaintext_size );
+    status = psa_aead_final_checks( operation );
 
     if( status != PSA_SUCCESS )
         goto exit;