2 files changed, 6 insertions, 12 deletions
diff --git a/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.c b/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.c
index a816d704e2..524d6c060d 100644
--- a/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.c
+++ b/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.c
@@ -68,12 +68,11 @@ UINTN                   mPdbNameModHandleArraySize = 0;
 **/
 EFI_STATUS
 EFIAPI
-DxeNt32PeCoffLibExtraActionConstructor (
-  IN EFI_HANDLE        ImageHandle,
-  IN EFI_SYSTEM_TABLE  *SystemTable
+Nt32PeCoffGetWinNtThunkStucture (
+  VOID
   )
 {
-	EFI_HOB_GUID_TYPE   *GuidHob;
+  EFI_HOB_GUID_TYPE   *GuidHob;
 
   //
   // Retrieve WinNtThunkProtocol from GUID'ed HOB
@@ -284,10 +283,10 @@ PeCoffLoaderRelocateImageExtraAction (
   ASSERT (ImageContext != NULL);
 
   if (mWinNt == NULL) {
-    return;
+    Nt32PeCoffGetWinNtThunkStucture ();
   }
 
-	//
+  //
   // If we load our own PE COFF images the Windows debugger can not source
   //  level debug our code. If a valid PDB pointer exists usw it to load
   //  the *.dll file as a library using Windows* APIs. This allows 
@@ -397,12 +396,9 @@ PeCoffLoaderUnloadImageExtraAction (
   IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext
   )
 {
-	VOID *ModHandle;
+  VOID *ModHandle;
 
   ASSERT (ImageContext != NULL);
-  if (mWinNt == NULL) {
-    return;
-  }
 
   ModHandle = RemoveModeHandle (ImageContext);
   if (ModHandle != NULL) {
diff --git a/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.inf b/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.inf
index f50e97f826..97a132b441 100644
--- a/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.inf
+++ b/Nt32Pkg/Library/DxeNt32PeCoffExtraActionLib/DxeNt32PeCoffExtraActionLib.inf
@@ -23,8 +23,6 @@
   VERSION_STRING                 = 1.0
   LIBRARY_CLASS                  = PeCoffExtraActionLib|DXE_CORE DXE_DRIVER DXE_RUNTIME_DRIVER UEFI_DRIVER
 
-  CONSTRUCTOR                    = DxeNt32PeCoffLibExtraActionConstructor
-
 #
 # The following information is for reference only and not required by the build tools.
 #