diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 0853c14..12285d3 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 (!PSA_KEY_LIFETIME_IS_VOLATILE(lifetime))
+    if (!PSA_KEY_LIFETIME_IS_VOLATILE(slot->attr.lifetime ))
     {
 #if defined(MBEDTLS_PSA_CRYPTO_SE_C)
         if( driver != NULL )