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