diff --git a/library/ssl_tls12_server.c b/library/ssl_tls12_server.c
index 36f8f16..e0b5aab 100644
--- a/library/ssl_tls12_server.c
+++ b/library/ssl_tls12_server.c
@@ -2886,7 +2886,7 @@
         status = psa_get_key_attributes( ssl->handshake->ecdh_psa_privkey,
                                          &key_attributes );
         if( status != PSA_SUCCESS)
-            return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
+            return( psa_ssl_status_to_mbedtls( status ) );
 
         ssl->handshake->ecdh_psa_type = psa_get_key_type( &key_attributes );
         ssl->handshake->ecdh_bits = psa_get_key_bits( &key_attributes );