\r
**/\r
\r
-#include <FrameworkDxe.h>\r
+//\r
+// The package level header files this module uses\r
+//\r
+#include <FrameworkSmm.h>\r
+//\r
+// The protocols, PPI and GUID defintions for this module\r
+//\r
+#include <Protocol/LoadedImage.h>\r
+#include <Protocol/SmmBase.h>\r
+#include <Protocol/DevicePath.h>\r
+//\r
+// The Library classes this module consumes\r
+//\r
+#include <Library/DxeSmmDriverEntryPoint.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/DebugLib.h>\r
-#include <Library/DxeSmmDriverEntryPoint.h>\r
\r
-#include <Protocol/LoadedImage.h>\r
-#include <Protocol/SmmBase.h>\r
\r
EFI_BOOT_SERVICES *mBS;\r
\r