//\r
// The package level header files this module uses\r
//\r
-#include <FrameworkDxe.h>\r
+#include <PiDxe.h>\r
+\r
+\r
//\r
// The protocols, PPI and GUID defintions for this module\r
//\r
#include <Library/UefiLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
-\r
+#include <Library/HobLib.h>\r
+#include <Common/FlashMap.h>\r
+#include <Guid/FlashMapHob.h>\r
\r
#endif\r