diff --git a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
index 70c71c6db4..de719f3e2e 100644
--- a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
+++ b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
@@ -262,7 +262,7 @@ Returns:
 {
   EFI_STATUS                         Status;
   EFI_GUID                           **DecompressGuidList;
-  UINTN                              DecompressMethodNumber;
+  UINT32                             DecompressMethodNumber;
 
   //
   // Install SEP to a new handle
diff --git a/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c b/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c
index e7359e5684..381d60cb2a 100644
--- a/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c
+++ b/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c
@@ -958,7 +958,8 @@ CustomDecompressExtractSection (
   EFI_STATUS      Status;
   UINT8           *ScratchBuffer;
   UINT32          ScratchSize;
-  UINT32          SectionLength;  
+  UINT32          SectionLength;
+  UINT32          DestinationSize;  
   
   //
   // Set authentic value to zero.
@@ -975,7 +976,7 @@ CustomDecompressExtractSection (
              (GUID *) ((UINT8 *) InputSection + sizeof (EFI_COMMON_SECTION_HEADER)),
              (UINT8 *) InputSection + sizeof (EFI_GUID_DEFINED_SECTION),
              SectionLength - sizeof (EFI_GUID_DEFINED_SECTION),
-             OutputSize,
+             &DestinationSize,
              &ScratchSize
              );
   if (EFI_ERROR (Status)) {
@@ -996,6 +997,7 @@ CustomDecompressExtractSection (
   //
   // Allocate destination buffer
   //
+  *OutputSize   = (UINTN) DestinationSize;
   *OutputBuffer = AllocatePages (EFI_SIZE_TO_PAGES (*OutputSize));
   if (*OutputBuffer == NULL) {
     return EFI_OUT_OF_RESOURCES;