diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index eb217f9..bc46ad2 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -574,7 +574,7 @@
                                     output, output_size, 
                                     &output_length) == PSA_SUCCESS );
 	TEST_ASSERT( psa_cipher_finish( &operation, output + output_length,
-                                    &output_length) == PSA_SUCCESS );
+                                    output_size, &output_length) == PSA_SUCCESS );
 
     TEST_ASSERT( psa_cipher_abort( &operation ) == PSA_SUCCESS );
 
@@ -637,7 +637,7 @@
                                     output, output_size, 
                                     &output_length) == PSA_SUCCESS );
 	TEST_ASSERT( psa_cipher_finish( &operation, output + output_length,
-                                    &output_length) == PSA_SUCCESS );
+                                    output_size, &output_length) == PSA_SUCCESS );
 
     TEST_ASSERT( psa_cipher_abort( &operation ) == PSA_SUCCESS );
 
@@ -700,7 +700,7 @@
                                     output1, output1_size, 
                                     &output1_length) == PSA_SUCCESS );
 	TEST_ASSERT( psa_cipher_finish( &operation1, output1 + output1_length,
-                                    &tmp_output_length) == PSA_SUCCESS );
+                                    output1_size, &tmp_output_length) == PSA_SUCCESS );
 									
     output1_length += tmp_output_length;
 
@@ -715,7 +715,7 @@
                                     output2, output2_size, &output2_length) == PSA_SUCCESS );
     tmp_output_length = 0;
 	TEST_ASSERT( psa_cipher_finish( &operation2, output2 + output2_length,
-                                    &tmp_output_length) == PSA_SUCCESS );  
+                                    output2_size, &tmp_output_length) == PSA_SUCCESS );  
     
 	output2_length += tmp_output_length;