diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index ca461c20e..68fa0ef67 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -27,6 +27,17 @@
 
 #if defined(MBEDTLS_PSA_CRYPTO_C)
 
+//! In case MBEDTLS_PSA_CRYPTO_SPM is defined the code is built for SPM (Secure
+//! Partition Manager) integration which separate the code into two parts
+//! NSPE (Non-Secure Process Environment) and SPE (Secure Process Environment).
+//! In this mode an additional header file should be included.
+#if defined(MBEDTLS_PSA_CRYPTO_SPM)
+//! PSA_CRYPTO_SECURE means that this file is compiled to the SPE side.
+//! some headers will be affected by this flag.
+#define PSA_CRYPTO_SECURE 1
+#include "crypto_spe.h"
+#endif
+
 #include "psa/crypto.h"
 
 #include <stdlib.h>
@@ -2481,7 +2492,7 @@ psa_status_t psa_cipher_abort( psa_cipher_operation_t *operation )
 /****************************************************************/
 /* Key Policy */
 /****************************************************************/
-
+#if !defined(MBEDTLS_PSA_CRYPTO_SPM)
 void psa_key_policy_init( psa_key_policy_t *policy )
 {
     memset( policy, 0, sizeof( *policy ) );
@@ -2504,7 +2515,7 @@ psa_algorithm_t psa_key_policy_get_algorithm( psa_key_policy_t *policy )
 {
     return( policy->alg );
 }
-
+#endif /* !defined(MBEDTLS_PSA_CRYPTO_SPM) */
 psa_status_t psa_set_key_policy( psa_key_slot_t key,
                                  const psa_key_policy_t *policy )
 {
diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index 9eac29b43..c90447f81 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -1,5 +1,10 @@
 /* BEGIN_HEADER */
 #include <stdint.h>
+
+#if defined(MBEDTLS_PSA_CRYPTO_SPM)
+#include "spm/psa_defs.h"
+#endif
+
 #include "mbedtls/asn1write.h"
 #include "psa/crypto.h"