diff --git a/tests/src/drivers/keygen.c b/tests/src/drivers/keygen.c
index d493ab3..f15a4bc 100644
--- a/tests/src/drivers/keygen.c
+++ b/tests/src/drivers/keygen.c
@@ -30,7 +30,7 @@
 #include "mbedtls/ecp.h"
 #include "mbedtls/error.h"
 
-#include "drivers/keygen.h"
+#include "test/drivers/keygen.h"
 
 #include "test/random.h"
 
diff --git a/tests/src/drivers/signature.c b/tests/src/drivers/signature.c
index 5299a96..028d24a 100644
--- a/tests/src/drivers/signature.c
+++ b/tests/src/drivers/signature.c
@@ -30,7 +30,7 @@
 #include "psa_crypto_core.h"
 #include "mbedtls/ecp.h"
 
-#include "drivers/signature.h"
+#include "test/drivers/signature.h"
 
 #include "mbedtls/md.h"
 #include "mbedtls/ecdsa.h"