diff --git a/tests/src/drivers/keygen.c b/tests/src/drivers/keygen.c
index 7f14b20..4f30f0e 100644
--- a/tests/src/drivers/keygen.c
+++ b/tests/src/drivers/keygen.c
@@ -25,6 +25,7 @@
 
 #if defined(MBEDTLS_PSA_CRYPTO_DRIVERS) && defined(PSA_CRYPTO_DRIVER_TEST)
 #include "psa/crypto.h"
+#include "psa_crypto_core.h"
 #include "mbedtls/ecp.h"
 #include "mbedtls/error.h"
 
diff --git a/tests/src/drivers/signature.c b/tests/src/drivers/signature.c
index 62ba407..04c5de4 100644
--- a/tests/src/drivers/signature.c
+++ b/tests/src/drivers/signature.c
@@ -25,6 +25,7 @@
 
 #if defined(MBEDTLS_PSA_CRYPTO_DRIVERS) && defined(PSA_CRYPTO_DRIVER_TEST)
 #include "psa/crypto.h"
+#include "psa_crypto_core.h"
 #include "mbedtls/ecp.h"
 
 #include "drivers/signature.h"