\r
typedef struct _PXEBC_PRIVATE_DATA PXEBC_PRIVATE_DATA;\r
\r
-#include <PiDxe.h>\r
+#include <Uefi.h>\r
\r
#include <Guid/SmBios.h>\r
#include <IndustryStandard/SmBios.h>\r
\r
#include <Library/DebugLib.h>\r
#include <Library/BaseMemoryLib.h>\r
+#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiDriverEntryPoint.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/UefiLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/NetLib.h>\r
+#include <Library/DpcLib.h>\r
\r
#include "PxeBcDriver.h"\r
#include "PxeArch.h"\r