diff --git a/Nt32Pkg/Library/DxeNt32PeCoffLib/DxeNt32PeCoffLib.c b/Nt32Pkg/Library/DxeNt32PeCoffLib/DxeNt32PeCoffLib.c
index b58b85599a..d8eb7c9436 100644
--- a/Nt32Pkg/Library/DxeNt32PeCoffLib/DxeNt32PeCoffLib.c
+++ b/Nt32Pkg/Library/DxeNt32PeCoffLib/DxeNt32PeCoffLib.c
@@ -218,7 +218,7 @@ PeCoffLoaderRelocateImageForRuntime (
 RETURN_STATUS
 EFIAPI
 PeCoffLoaderUnloadImage (
-  IN PE_COFF_LOADER_IMAGE_CONTEXT         *ImageContext
+  IN OUT PE_COFF_LOADER_IMAGE_CONTEXT         *ImageContext
   )
 {
   return mPeiEfiPeiPeCoffLoader->UnloadImage (mPeiEfiPeiPeCoffLoader, ImageContext);
diff --git a/Nt32Pkg/Library/PeiCoreNt32PeCoffLib/PeiCoreNt32PeCoffLib.c b/Nt32Pkg/Library/PeiCoreNt32PeCoffLib/PeiCoreNt32PeCoffLib.c
index e4e6225193..ea1276fec9 100644
--- a/Nt32Pkg/Library/PeiCoreNt32PeCoffLib/PeiCoreNt32PeCoffLib.c
+++ b/Nt32Pkg/Library/PeiCoreNt32PeCoffLib/PeiCoreNt32PeCoffLib.c
@@ -252,7 +252,7 @@ PeCoffLoaderRelocateImageForRuntime (
 RETURN_STATUS
 EFIAPI
 PeCoffLoaderUnloadImage (
-  IN PE_COFF_LOADER_IMAGE_CONTEXT         *ImageContext
+  IN OUT PE_COFF_LOADER_IMAGE_CONTEXT         *ImageContext
   )
 {
   if (mPeiEfiPeiPeCoffLoader == NULL) {
diff --git a/Nt32Pkg/Library/PeiNt32PeCoffLib/PeiNt32PeCoffLib.c b/Nt32Pkg/Library/PeiNt32PeCoffLib/PeiNt32PeCoffLib.c
index 997585cd46..91ce7f6468 100644
--- a/Nt32Pkg/Library/PeiNt32PeCoffLib/PeiNt32PeCoffLib.c
+++ b/Nt32Pkg/Library/PeiNt32PeCoffLib/PeiNt32PeCoffLib.c
@@ -248,7 +248,7 @@ PeCoffLoaderRelocateImageForRuntime (
 RETURN_STATUS
 EFIAPI
 PeCoffLoaderUnloadImage (
-  IN PE_COFF_LOADER_IMAGE_CONTEXT         *ImageContext
+  IN OUT PE_COFF_LOADER_IMAGE_CONTEXT         *ImageContext
   )
 {
   return mPeiEfiPeiPeCoffLoader->UnloadImage (mPeiEfiPeiPeCoffLoader, ImageContext);