diff --git a/library/psa_crypto_cipher.c b/library/psa_crypto_cipher.c
index 6206237..36c7e73 100644
--- a/library/psa_crypto_cipher.c
+++ b/library/psa_crypto_cipher.c
@@ -392,7 +392,7 @@
                                    input_length, output, output_length ) );
 
         if( *output_length > output_size )
-            return PSA_ERROR_CORRUPTION_DETECTED;
+            return( PSA_ERROR_CORRUPTION_DETECTED );
     }
 
     return( status );
@@ -492,8 +492,7 @@
     if( status != PSA_SUCCESS )
         goto exit;
 
-    accumulated_length += olength;
-    *output_length = accumulated_length;
+    *output_length = accumulated_length + olength;
 
 exit:
     if( status == PSA_SUCCESS )
@@ -542,8 +541,7 @@
     if( status != PSA_SUCCESS )
         goto exit;
 
-    accumulated_length += olength;
-    *output_length = accumulated_length;
+    *output_length = accumulated_length + olength;
 
 exit:
     if ( status == PSA_SUCCESS )