diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index 451e232..59a96d2 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -27,6 +27,7 @@
 
 #if defined(MBEDTLS_VERSION_C)
 #include "mbedtls/build_info.h"
+#include "mbedtls/version.h"
 #endif /* MBEDTLS_VERSION_C */
 
 #if defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)
@@ -1004,7 +1005,7 @@
             {
                 mbedtls_printf( "build version: %s (build %u)\n",
                                 MBEDTLS_VERSION_STRING,
-                                MBEDTLS_VERSION_NUMBER );
+                                mbedtls_version_get_number() );
                 goto exit;
             }
         }
@@ -2481,7 +2482,7 @@
 
 #if defined(MBEDTLS_VERSION_C)
     mbedtls_printf( "build version: %s (build %u)\n",
-                    MBEDTLS_VERSION_STRING, MBEDTLS_VERSION_NUMBER );
+                    MBEDTLS_VERSION_STRING, mbedtls_version_get_number() );
 #endif /* MBEDTLS_VERSION_C */
 
 #if defined(MBEDTLS_X509_CRT_PARSE_C)