diff --git a/DuetPkg/CpuDxe/CpuDxe.h b/DuetPkg/CpuDxe/CpuDxe.h
index 74f1eeba4f..3fd1254e90 100644
--- a/DuetPkg/CpuDxe/CpuDxe.h
+++ b/DuetPkg/CpuDxe/CpuDxe.h
@@ -18,7 +18,7 @@ Abstract:
 #ifndef _CPU_DXE_H
 #define _CPU_DXE_H
 
-#include <PiDxe.h>
+#include <FrameworkDxe.h>
 
 #include <Protocol/Cpu.h>
 #include <Protocol/Legacy8259.h>