Merge the PI enabling works from the branch

[mirror_edk2.git] / MdeModulePkg / Universal / MemoryTest / BaseMemoryTestPei / BaseMemoryTest.c

diff --git a/MdeModulePkg/Universal/MemoryTest/BaseMemoryTestPei/BaseMemoryTest.c b/MdeModulePkg/Universal/MemoryTest/BaseMemoryTestPei/BaseMemoryTest.c
index f1ab82d87abb29eb3fc5d040b563f95557070491..f51869c7733204298071b9e94e73fbbc654ab906 100644 (file)
--- a/MdeModulePkg/Universal/MemoryTest/BaseMemoryTestPei/BaseMemoryTest.c
+++ b/MdeModulePkg/Universal/MemoryTest/BaseMemoryTestPei/BaseMemoryTest.c
@@ -20,6 +20,7 @@ Abstract:
 --*/\r
 \r
 #include <BaseMemoryTest.h>\r
+#include <Library/PeiServicesLib.h>\r
 \r
 static PEI_BASE_MEMORY_TEST_PPI mPeiBaseMemoryTestPpi = { BaseMemoryTest };\r
 \r
@@ -51,11 +52,9 @@ Returns:
 \r
 --*/  \r
 {\r
-  EFI_STATUS  Status;\r
 \r
-  Status = (**PeiServices).InstallPpi (PeiServices, &PpiListPeiBaseMemoryTest);\r
-\r
-  return Status;\r
+  return PeiServicesInstallPpi (&PpiListPeiBaseMemoryTest);\r
+  \r
 }\r
 \r
 EFI_STATUS\r