diff --git a/app/CMakeLists.txt b/app/CMakeLists.txt
index d71ad6f..9b853cd 100755
--- a/app/CMakeLists.txt
+++ b/app/CMakeLists.txt
@@ -174,6 +174,8 @@
             set(CPU_ARCH armv8m_ml)
         elseif (${TFM_SYSTEM_ARCHITECTURE} STREQUAL armv8-m.base)
             set(CPU_ARCH armv8m_bl)
+        elseif (${TFM_SYSTEM_ARCHITECTURE} STREQUAL armv8.1-m.main)
+            set(CPU_ARCH armv81m_ml)
         elseif (${TFM_SYSTEM_ARCHITECTURE} STREQUAL armv7-m)
             set(CPU_ARCH armv7m)
         endif()