diff --git a/ntoskrnl/io/pnpmgr/plugplay.c b/ntoskrnl/io/pnpmgr/plugplay.c
index b9551207c33..4d64c1e54da 100644
--- a/ntoskrnl/io/pnpmgr/plugplay.c
+++ b/ntoskrnl/io/pnpmgr/plugplay.c
@@ -26,7 +26,8 @@ static KEVENT IopPnpNotifyEvent;
 
 /* FUNCTIONS *****************************************************************/
 
-NTSTATUS CODE_SEG("INIT")
+CODE_SEG("INIT")
+NTSTATUS
 IopInitPlugPlayEvents(VOID)
 {
     InitializeListHead(&IopPnpEventQueueHead);
diff --git a/ntoskrnl/mm/amd64/init.c b/ntoskrnl/mm/amd64/init.c
index 0c6a868b144..38579414c6a 100644
--- a/ntoskrnl/mm/amd64/init.c
+++ b/ntoskrnl/mm/amd64/init.c
@@ -440,9 +440,9 @@ MiSetupPfnForPageTable(
     Pfn->u2.ShareCount++;
 }
 
+CODE_SEG("INIT")
 static
 VOID
-CODE_SEG("INIT")
 MiBuildPfnDatabaseFromPageTables(VOID)
 {
     PVOID Address = NULL;