diff --git a/tests/suites/test_suite_md.function b/tests/suites/test_suite_md.function
index 2768707e9f..75087e7eea 100644
--- a/tests/suites/test_suite_md.function
+++ b/tests/suites/test_suite_md.function
@@ -1,5 +1,6 @@
 /* BEGIN_HEADER */
 #include "mbedtls/md.h"
+#include "../library/md_psa.h"
 /* END_HEADER */
 
 /* BEGIN_DEPENDENCIES