1 files changed, 7 insertions, 2 deletions
diff --git a/MdeModulePkg/Core/Dxe/Image/Image.c b/MdeModulePkg/Core/Dxe/Image/Image.c
index 0719d4b690..00f6bdced5 100644
--- a/MdeModulePkg/Core/Dxe/Image/Image.c
+++ b/MdeModulePkg/Core/Dxe/Image/Image.c
@@ -997,8 +997,12 @@ CoreLoadImage (
   )
 {
   EFI_STATUS    Status;
+  UINT64        Tick;
 
-  PERF_START (NULL, "LoadImage", NULL, 0);
+  Tick = 0;
+  PERF_CODE (
+    Tick = GetPerformanceCounter ();
+  );
 
   Status = CoreLoadImageCommon (
              BootPolicy,
@@ -1013,7 +1017,8 @@ CoreLoadImage (
              EFI_LOAD_PE_IMAGE_ATTRIBUTE_RUNTIME_REGISTRATION | EFI_LOAD_PE_IMAGE_ATTRIBUTE_DEBUG_IMAGE_INFO_TABLE_REGISTRATION
              );
 
-  PERF_END (NULL, "LoadImage", NULL, 0);
+  PERF_START (*ImageHandle, LOAD_IMAGE_TOK, NULL, Tick);
+  PERF_END (*ImageHandle, LOAD_IMAGE_TOK, NULL, 0);
 
   return Status;
 }