diff --git a/Nt32Pkg/Library/PeiNt32OemHookStatusCodeLib/Nt32OemHookStatusCodeLib.c b/Nt32Pkg/Library/PeiNt32OemHookStatusCodeLib/Nt32OemHookStatusCodeLib.c
index bc8db38796..d7a1fb021a 100644
--- a/Nt32Pkg/Library/PeiNt32OemHookStatusCodeLib/Nt32OemHookStatusCodeLib.c
+++ b/Nt32Pkg/Library/PeiNt32OemHookStatusCodeLib/Nt32OemHookStatusCodeLib.c
@@ -17,7 +17,7 @@
 //
 // The package level header files this module uses
 //
-#include <PiPei.h>
+#include <FrameworkPei.h>
 #include <WinNtPeim.h>
 
 //
diff --git a/Nt32Pkg/Sec/SecMain.h b/Nt32Pkg/Sec/SecMain.h
index 3d2dd2c53e..82bb29d0ea 100644
--- a/Nt32Pkg/Sec/SecMain.h
+++ b/Nt32Pkg/Sec/SecMain.h
@@ -19,9 +19,7 @@ Abstract:
 **/
 
 #include <stdio.h>
-#include <Base.h>
 #include <PiPei.h>
-#include <PiDxe.h>
 #include <WinNtPeim.h>
 #include <Guid/StatusCodeDataTypeDebug.h>
 #include <Library/BaseLib.h>