diff --git a/tests/include/test/psa_crypto_helpers.h b/tests/include/test/psa_crypto_helpers.h
index c29134a..55f24e0 100644
--- a/tests/include/test/psa_crypto_helpers.h
+++ b/tests/include/test/psa_crypto_helpers.h
@@ -21,7 +21,7 @@
 #include "mbedtls/psa_util.h"
 #endif
 
-#if defined(MBEDTLS_TEST_HOOKS)
+#if defined(MBEDTLS_TEST_HOOKS) && defined(MBEDTLS_PSA_CRYPTO_C)
 #include "test/psa_memory_poisoning_wrappers.h"
 #endif
 
diff --git a/tests/src/helpers.c b/tests/src/helpers.c
index 41f805e..27a7a7a 100644
--- a/tests/src/helpers.c
+++ b/tests/src/helpers.c
@@ -17,7 +17,7 @@
 #include <test/psa_crypto_helpers.h>
 #endif
 
-#if defined(MBEDTLS_TEST_HOOKS)
+#if defined(MBEDTLS_TEST_HOOKS) && defined(MBEDTLS_PSA_CRYPTO_C)
 #include <test/psa_memory_poisoning_wrappers.h>
 #endif