diff --git a/programs/pkey/dh_client.c b/programs/pkey/dh_client.c
index 546b02e..0b340d5 100644
--- a/programs/pkey/dh_client.c
+++ b/programs/pkey/dh_client.c
@@ -127,7 +127,7 @@
         goto exit;
     }
 
-    rsa.len = ( mbedtls_mpi_msb( &rsa.N ) + 7 ) >> 3;
+    rsa.len = ( mbedtls_mpi_bitlen( &rsa.N ) + 7 ) >> 3;
 
     fclose( f );