diff --git a/library/ssl_tls13_client.c b/library/ssl_tls13_client.c
index 097505a..a6b3c54 100644
--- a/library/ssl_tls13_client.c
+++ b/library/ssl_tls13_client.c
@@ -677,9 +677,9 @@
     mbedtls_ssl_session *session = ssl->session_negotiate;
     return session != NULL &&
            mbedtls_ssl_conf_tls13_check_kex_modes(ssl,
-                                                  mbedtls_ssl_tls13_session_get_ticket_flags(
+                                                  mbedtls_ssl_session_get_ticket_flags(
                                                       session,
-                                                      MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_ALL));
+                                                      MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_ALL));
 }
 
 static int ssl_tls13_has_configured_ticket(mbedtls_ssl_context *ssl)
@@ -2630,7 +2630,7 @@
     session->ticket_len = ticket_len;
 
     /* Clear all flags in ticket_flags */
-    mbedtls_ssl_tls13_session_clear_ticket_flags(
+    mbedtls_ssl_session_clear_ticket_flags(
         session, MBEDTLS_SSL_TLS1_3_TICKET_FLAGS_MASK);
 
     MBEDTLS_SSL_CHK_BUF_READ_PTR(p, end, 2);
@@ -2717,7 +2717,7 @@
                           session->resumption_key_len);
 
     /* Set ticket_flags depends on the selected key exchange modes */
-    mbedtls_ssl_tls13_session_set_ticket_flags(
+    mbedtls_ssl_session_set_ticket_flags(
         session, ssl->conf->tls13_kex_modes);
     MBEDTLS_SSL_DEBUG_TICKET_FLAGS(4, session->ticket_flags);