diff --git a/mm/kasan/CMakeLists.txt b/mm/kasan/CMakeLists.txt
index d33e1805fed..d802a55b1a4 100644
--- a/mm/kasan/CMakeLists.txt
+++ b/mm/kasan/CMakeLists.txt
@@ -34,4 +34,9 @@ if(CONFIG_MM_KASAN)
 endif()
 
 target_sources(mm PRIVATE ${SRCS})
-target_compile_options(mm PRIVATE ${FLAGS})
+
+if(CONFIG_BUILD_FLAT)
+  target_compile_options(mm PRIVATE ${FLAGS})
+else()
+  target_compile_options(kmm PRIVATE ${FLAGS})
+endif()