diff --git a/library/psa_crypto_mac.c b/library/psa_crypto_mac.c
index cf2f072..19671ec 100644
--- a/library/psa_crypto_mac.c
+++ b/library/psa_crypto_mac.c
@@ -600,7 +600,7 @@
 }
 
 psa_status_t mbedtls_transparent_test_driver_mac_sign_setup(
-    mbedtls_transparent_test_driver_mac_operation_t *operation,
+    mbedtls_psa_mac_operation_t *operation,
     const psa_key_attributes_t *attributes,
     const uint8_t *key_buffer,
     size_t key_buffer_size,
@@ -614,7 +614,7 @@
 }
 
 psa_status_t mbedtls_transparent_test_driver_mac_verify_setup(
-    mbedtls_transparent_test_driver_mac_operation_t *operation,
+    mbedtls_psa_mac_operation_t *operation,
     const psa_key_attributes_t *attributes,
     const uint8_t *key_buffer,
     size_t key_buffer_size,
@@ -628,7 +628,7 @@
 }
 
 psa_status_t mbedtls_transparent_test_driver_mac_update(
-    mbedtls_transparent_test_driver_mac_operation_t *operation,
+    mbedtls_psa_mac_operation_t *operation,
     const uint8_t *input,
     size_t input_length )
 {
@@ -639,7 +639,7 @@
 }
 
 psa_status_t mbedtls_transparent_test_driver_mac_sign_finish(
-    mbedtls_transparent_test_driver_mac_operation_t *operation,
+    mbedtls_psa_mac_operation_t *operation,
     uint8_t *mac,
     size_t mac_size,
     size_t *mac_length )
@@ -651,7 +651,7 @@
 }
 
 psa_status_t mbedtls_transparent_test_driver_mac_verify_finish(
-    mbedtls_transparent_test_driver_mac_operation_t *operation,
+    mbedtls_psa_mac_operation_t *operation,
     const uint8_t *mac,
     size_t mac_length )
 {
@@ -662,7 +662,7 @@
 }
 
 psa_status_t mbedtls_transparent_test_driver_mac_abort(
-    mbedtls_transparent_test_driver_mac_operation_t *operation )
+    mbedtls_psa_mac_operation_t *operation )
 {
     return( mac_abort( operation ) );
 }