diff --git a/library/kani_core/src/mem_init.rs b/library/kani_core/src/mem_init.rs
index ecafe5d8e861..935f14d71c5d 100644
--- a/library/kani_core/src/mem_init.rs
+++ b/library/kani_core/src/mem_init.rs
@@ -23,7 +23,7 @@
 macro_rules! kani_mem_init {
     ($core:path) => {
         /// Global object for tracking memory initialization state.
-        #[kanitool::fn_marker = "MemoryInitializationStateModel"]
+        #[rustc_diagnostic_item = "KaniMemoryInitializationState"]
         static mut MEM_INIT_STATE: MemoryInitializationState = MemoryInitializationState::new();
 
         /// Global object for tracking union initialization state across function boundaries.