diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index bb046ef..0853c14 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1647,7 +1647,7 @@
     (void) driver;
 
 #if defined(MBEDTLS_PSA_CRYPTO_STORAGE_C)
-    if( slot->attr.lifetime != PSA_KEY_LIFETIME_VOLATILE )
+    if (!PSA_KEY_LIFETIME_IS_VOLATILE(lifetime))
     {
 #if defined(MBEDTLS_PSA_CRYPTO_SE_C)
         if( driver != NULL )