diff --git a/tests/src/certs.c b/tests/src/certs.c
index 5099577..bd5c49e 100644
--- a/tests/src/certs.c
+++ b/tests/src/certs.c
@@ -25,7 +25,7 @@
 
 #include "mbedtls/legacy_or_psa.h"
 
-#include "pk_wrap.c"
+#include "mbedtls/pk.h"
 
 /*
  * Test CA Certificates
diff --git a/tests/suites/test_suite_debug.function b/tests/suites/test_suite_debug.function
index 8d3f6e7..b5d077c 100644
--- a/tests/suites/test_suite_debug.function
+++ b/tests/suites/test_suite_debug.function
@@ -2,7 +2,7 @@
 #include "mbedtls/debug.h"
 #include "string.h"
 #include "mbedtls/legacy_or_psa.h"
-#include "pk_wrap.h"
+#include "pk.h"
 
 struct buffer_data {
     char buf[2000];
diff --git a/tests/suites/test_suite_ssl.function b/tests/suites/test_suite_ssl.function
index e3d5705..e67c3d1 100644
--- a/tests/suites/test_suite_ssl.function
+++ b/tests/suites/test_suite_ssl.function
@@ -6,7 +6,7 @@
 #include <ssl_tls13_keys.h>
 #include <ssl_tls13_invasive.h>
 #include "test/certs.h"
-#include "pk_wrap.h"
+#include "pk.h"
 
 #if defined(MBEDTLS_SSL_CACHE_C)
 #include "mbedtls/ssl_cache.h"