diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index 4494291..376f6cf 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -1677,6 +1677,7 @@
             return MBEDTLS_ERR_SSL_BAD_INPUT_DATA;
         }
         ssl->handshake->ciphersuite_info = ciphersuite_info;
+        ssl->handshake->key_exchange_mode = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_ALL;
     }
 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
 
@@ -1686,7 +1687,6 @@
     }
 
     ssl->handshake->resume = 1;
-    ssl->handshake->key_exchange_mode = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_ALL;
 
     return 0;
 }