diff --git a/programs/pkey/dh_server.c b/programs/pkey/dh_server.c
index 81c0c9f..8fda29f 100644
--- a/programs/pkey/dh_server.c
+++ b/programs/pkey/dh_server.c
@@ -134,7 +134,7 @@
         goto exit;
     }
 
-    rsa.len = ( mbedtls_mpi_msb( &rsa.N ) + 7 ) >> 3;
+    rsa.len = ( mbedtls_mpi_bitlen( &rsa.N ) + 7 ) >> 3;
 
     fclose( f );