diff --git a/UefiCpuPkg/PiSmmCpuDxeSmm/X64/SmmProfileArch.c b/UefiCpuPkg/PiSmmCpuDxeSmm/X64/SmmProfileArch.c
index 79e23ef647..065fb2c24c 100644
--- a/UefiCpuPkg/PiSmmCpuDxeSmm/X64/SmmProfileArch.c
+++ b/UefiCpuPkg/PiSmmCpuDxeSmm/X64/SmmProfileArch.c
@@ -78,7 +78,7 @@ InitPagesForPFHandler (
   //
   Address = NULL;
   Address = AllocatePages (MAX_PF_PAGE_COUNT);
-  ASSERT_EFI_ERROR (Address != NULL);
+  ASSERT (Address != NULL);
 
   mPFPageBuffer =  (UINT64)(UINTN) Address;
   mPFPageIndex = 0;