diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index 062ff25..c1436c5 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -8233,7 +8233,7 @@
     }
 
     if( input_offset != len )
-        return PSA_ERROR_INVALID_ARGUMENT;
+        return MBEDTLS_ERR_SSL_HANDSHAKE_FAILURE;
 
     return( 0 );
 }