diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index eb6fae0..d8de189 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -3535,7 +3535,7 @@
       .core = slot->attr
     };
 
-    status = mbedtls_psa_aead_encrypt(
+    status = psa_driver_wrapper_aead_encrypt(
         &attributes, slot->key.data, slot->key.bytes,
         alg,
         nonce, nonce_length,
@@ -3577,7 +3577,7 @@
       .core = slot->attr
     };
 
-    status = mbedtls_psa_aead_decrypt(
+    status = psa_driver_wrapper_aead_decrypt(
         &attributes, slot->key.data, slot->key.bytes,
         alg,
         nonce, nonce_length,