diff --git a/cmsis/TARGET_CORTEX_M/mbed_tz_context.c b/cmsis/TARGET_CORTEX_M/mbed_tz_context.c
index e2e82942f8..8e9541f75d 100644
--- a/cmsis/TARGET_CORTEX_M/mbed_tz_context.c
+++ b/cmsis/TARGET_CORTEX_M/mbed_tz_context.c
@@ -22,6 +22,8 @@
  * limitations under the License.
  */
 
+#if defined (__ARM_FEATURE_CMSE) &&  (__ARM_FEATURE_CMSE == 3U)
+
 #include "RTE_Components.h"
 #include CMSIS_device_header
 #include "tz_context.h"
@@ -198,3 +200,4 @@ uint32_t TZ_StoreContext_S (TZ_MemoryId_t id) {
 
   return 1U;    // Success
 }
+#endif