diff --git a/library/psa_crypto_driver_wrappers.c b/library/psa_crypto_driver_wrappers.c
index ce49a22..1e17435 100644
--- a/library/psa_crypto_driver_wrappers.c
+++ b/library/psa_crypto_driver_wrappers.c
@@ -1330,8 +1330,7 @@
                         key_buffer, key_buffer_size,
                         alg );
 
-            if( status == PSA_SUCCESS )
-                operation->id = PSA_CRYPTO_MBED_TLS_DRIVER_ID;
+            operation->id = PSA_CRYPTO_MBED_TLS_DRIVER_ID;
 
             return( status );
 
@@ -1382,8 +1381,7 @@
                         key_buffer, key_buffer_size,
                         alg );
 
-            if( status == PSA_SUCCESS )
-                operation->id = PSA_CRYPTO_MBED_TLS_DRIVER_ID;
+            operation->id = PSA_CRYPTO_MBED_TLS_DRIVER_ID;
 
             return( status );