diff --git a/library/pk.c b/library/pk.c
index 462807c..f2c1ad5 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -24,7 +24,7 @@
 #include "pk_wrap.h"
 #include "pkwrite.h"
 
-#include "md_internal.h"
+#include "hash_info.h"
 
 #include "mbedtls/platform_util.h"
 #include "mbedtls/error.h"
@@ -363,7 +363,7 @@
     if( *hash_len != 0 )
         return( 0 );
 
-    *hash_len = mbedtls_md_internal_get_size( md_alg );
+    *hash_len = mbedtls_hash_info_get_size( md_alg );
 
     if( *hash_len == 0 )
         return( -1 );