diff --git a/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.h b/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.h
index 034da90939..21ad360352 100644
--- a/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.h
+++ b/MdeModulePkg/Bus/Ata/AtaBusDxe/AtaBus.h
@@ -18,7 +18,7 @@
 #ifndef _ATA_BUS_H_
 #define _ATA_BUS_H_
 
-#include <PiDxe.h>
+#include <Uefi.h>
 
 #include <Protocol/AtaPassThru.h>
 #include <Protocol/BlockIo.h>