Switch outbound back to handshake key after end_of_early_data

Signed-off-by: Xiaokang Qian <xiaokang.qian@arm.com>
diff --git a/library/ssl_tls13_client.c b/library/ssl_tls13_client.c
index 418c84d..4c07a64 100644
--- a/library/ssl_tls13_client.c
+++ b/library/ssl_tls13_client.c
@@ -2185,6 +2185,10 @@
 
         MBEDTLS_SSL_PROC_CHK(
             mbedtls_ssl_finish_handshake_msg(ssl, buf_len, 0));
+
+        /* Switch outbound back to handshake key after end_of_early_data */
+        mbedtls_ssl_set_outbound_transform(
+            ssl, ssl->handshake->transform_handshake);
     }
 
     MBEDTLS_SSL_PROC_CHK(