\r
**/\r
\r
-#include <PiDxe.h>\r
-#include <Library/BaseLib.h>\r
-#include <Library/EblAddExternalCommandLib.h>\r
-#include <Library/UefiBootServicesTableLib.h>\r
+#include <Uefi.h>\r
#include <Library/UefiLib.h>\r
-\r
+#include <Library/UefiBootServicesTableLib.h>\r
+#include <Library/EblAddExternalCommandLib.h>\r
+#include <Protocol/EblAddCommand.h>\r
\r
STATIC BOOLEAN gInstalledCommand = FALSE;\r
STATIC EFI_EVENT mEblCommandRegistration = NULL;\r