diff --git a/library/ssl_tls13_keys.c b/library/ssl_tls13_keys.c
index 429d124..aee2a85 100644
--- a/library/ssl_tls13_keys.c
+++ b/library/ssl_tls13_keys.c
@@ -148,7 +148,7 @@
     unsigned char null_salt[PSA_MAC_MAX_SIZE] = { '\0' };
     mbedtls_svc_key_id_t key = MBEDTLS_SVC_KEY_ID_INIT;
     psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
-    psa_status_t ret = MBEDTLS_ERR_SSL_INTERNAL_ERROR;
+    psa_status_t ret = PSA_ERROR_CORRUPTION_DETECTED;
 
     if( salt == NULL || salt_len == 0 )
     {
@@ -156,14 +156,14 @@
 
         if( salt_len != 0 )
         {
-            return( MBEDTLS_ERR_SSL_BAD_INPUT_DATA );
+            return( PSA_ERROR_INVALID_ARGUMENT );
         }
 
         hash_len = PSA_HASH_LENGTH( alg );
 
         if( hash_len == 0 )
         {
-            return( MBEDTLS_ERR_SSL_BAD_INPUT_DATA );
+            return( PSA_ERROR_INVALID_ARGUMENT );
         }
 
         salt = null_salt;