diff --git a/library/ssl_srv.c b/library/ssl_srv.c
index a31eda4..413dfc2 100644
--- a/library/ssl_srv.c
+++ b/library/ssl_srv.c
@@ -4088,11 +4088,11 @@
         goto cleanup;
     }
 
-    SSL_PROC_CHK( ssl_client_key_exchange_parse( ssl, ssl->in_msg,
+    MBEDTLS_SSL_CHK( ssl_client_key_exchange_parse( ssl, ssl->in_msg,
                                                  ssl->in_hslen ) );
 
     /* Update state */
-    SSL_PROC_CHK( ssl_client_key_exchange_postprocess( ssl ) );
+    MBEDTLS_SSL_CHK( ssl_client_key_exchange_postprocess( ssl ) );
 
 cleanup: