diff --git a/library/ssl_cli.c b/library/ssl_cli.c
index 41e19eb..36153a9 100644
--- a/library/ssl_cli.c
+++ b/library/ssl_cli.c
@@ -3238,7 +3238,8 @@
         unsigned char *own_pubkey_ecpoint;
         size_t own_pubkey_ecpoint_len;
 
-        psa_crypto_generator_t generator = PSA_CRYPTO_GENERATOR_INIT;
+        psa_key_derivation_operation_t generator =
+            PSA_KEY_DERIVATION_OPERATION_INIT;
 
         header_len = 4;
 
diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index d04fa9b..3f9818c 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -632,7 +632,7 @@
     psa_algorithm_t alg;
     psa_key_policy_t policy;
     psa_key_handle_t master_slot;
-    psa_crypto_generator_t generator = PSA_CRYPTO_GENERATOR_INIT;
+    psa_crypto_generator_t generator = PSA_KEY_DERIVATION_OPERATION_INIT;
 
     if( ( status = psa_allocate_key( &master_slot ) ) != PSA_SUCCESS )
         return( MBEDTLS_ERR_SSL_HW_ACCEL_FAILED );
@@ -1108,7 +1108,7 @@
             /* Perform PSK-to-MS expansion in a single step. */
             psa_status_t status;
             psa_algorithm_t alg;
-            psa_crypto_generator_t generator = PSA_CRYPTO_GENERATOR_INIT;
+            psa_crypto_generator_t generator = PSA_KEY_DERIVATION_OPERATION_INIT;
             psa_key_handle_t psk;
 
             MBEDTLS_SSL_DEBUG_MSG( 2, ( "perform PSA-based PSK-to-MS expansion" ) );