#include <Library/PeiServicesTablePointerLib.h>\r
#include <Library/DebugLib.h>\r
\r
-#include <Ppi/EmuPeiServicesTableUpdate.h>\r
+#include <Ppi/MemoryDiscovered.h>\r
\r
\r
CONST EFI_PEI_SERVICES **gPeiServices = NULL;\r
\r
EFI_PEI_NOTIFY_DESCRIPTOR mNotifyOnThunkList = {\r
(EFI_PEI_PPI_DESCRIPTOR_NOTIFY_CALLBACK | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
- &gEmuPeiServicesTableUpdatePpiGuid,\r
+ &gEfiPeiMemoryDiscoveredPpiGuid,\r
PeiServicesTablePointerNotifyCallback \r
};\r
\r
{\r
EFI_STATUS Status;\r
\r
+ gPeiServices = (CONST EFI_PEI_SERVICES **)PeiServices;\r
+\r
// register to be told when PeiServices pointer is updated\r
Status = (*PeiServices)->NotifyPpi (PeiServices, &mNotifyOnThunkList);\r
ASSERT_EFI_ERROR (Status);\r