diff --git a/library/ssl_cli.c b/library/ssl_cli.c
index e69aa27..46456c1 100644
--- a/library/ssl_cli.c
+++ b/library/ssl_cli.c
@@ -3320,11 +3320,11 @@
                                                   ssl->handshake->pmslen );
         if( status != PSA_SUCCESS )
         {
-            psa_generator_abort( &generator );
+            psa_key_derivation_abort( &generator );
             return( MBEDTLS_ERR_SSL_HW_ACCEL_FAILED );
         }
 
-        status = psa_generator_abort( &generator );
+        status = psa_key_derivation_abort( &generator );
         if( status != PSA_SUCCESS )
             return( MBEDTLS_ERR_SSL_HW_ACCEL_FAILED );