diff --git a/library/ssl_misc.h b/library/ssl_misc.h
index f9a4767..549162c 100644
--- a/library/ssl_misc.h
+++ b/library/ssl_misc.h
@@ -55,7 +55,7 @@
 #include "mbedtls/ecjpake.h"
 #endif
 
-#include "pk_wrap.h"
+#include "mbedtls/pk.h"
 #include "common.h"
 
 /* Shorthand for restartable ECC */
diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index e1d944c..0f7e61b 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -52,8 +52,6 @@
 #include "mbedtls/oid.h"
 #endif
 
-#include "pk_wrap.h"
-
 #if defined(MBEDTLS_TEST_HOOKS)
 static mbedtls_ssl_chk_buf_ptr_args chk_buf_ptr_fail_args;