diff --git a/library/ssl_cli.c b/library/ssl_cli.c
index 36153a9..882dc71 100644
--- a/library/ssl_cli.c
+++ b/library/ssl_cli.c
@@ -3301,11 +3301,12 @@
         content_len = own_pubkey_ecpoint_len + 1;
 
         /* Compute ECDH shared secret. */
-        status = psa_key_agreement( &generator,
-                                    handshake->ecdh_psa_privkey,
-                                    handshake->ecdh_psa_peerkey,
-                                    handshake->ecdh_psa_peerkey_len,
-                                    PSA_ALG_ECDH( PSA_ALG_SELECT_RAW ) );
+        status = psa_key_derivation_key_agreement(
+                &generator,
+                handshake->ecdh_psa_privkey,
+                handshake->ecdh_psa_peerkey,
+                handshake->ecdh_psa_peerkey_len,
+                PSA_ALG_ECDH( PSA_ALG_SELECT_RAW ) );
         if( status != PSA_SUCCESS )
             return( MBEDTLS_ERR_SSL_HW_ACCEL_FAILED );