diff --git a/ArmPlatformPkg/Library/ArmShellCmdRunAxf/ArmShellCmdRunAxf.h b/ArmPlatformPkg/Library/ArmShellCmdRunAxf/ArmShellCmdRunAxf.h
index 8599961d37..2708b53cc1 100644
--- a/ArmPlatformPkg/Library/ArmShellCmdRunAxf/ArmShellCmdRunAxf.h
+++ b/ArmPlatformPkg/Library/ArmShellCmdRunAxf/ArmShellCmdRunAxf.h
@@ -15,10 +15,8 @@
 #ifndef __ARM_SHELL_CMD_RUNAXF__
 #define __ARM_SHELL_CMD_RUNAXF__
 
-#include <ShellBase.h>
-
-#include <Protocol/EfiShell.h>
-#include <Protocol/EfiShellDynamicCommand.h>
+#include <Protocol/Shell.h>
+#include <Protocol/ShellDynamicCommand.h>
 
 #include <Library/HiiLib.h>
 #include <Library/ShellLib.h>