diff --git a/compiler/plugins/target/AMD-AIE/aie/AIEDialect.cpp b/compiler/plugins/target/AMD-AIE/aie/AIEDialect.cpp
index 26a040c3e..af05998aa 100644
--- a/compiler/plugins/target/AMD-AIE/aie/AIEDialect.cpp
+++ b/compiler/plugins/target/AMD-AIE/aie/AIEDialect.cpp
@@ -6,6 +6,8 @@
 
 #include "AIEDialect.h"
 
+#include "llvm/ADT/DenseSet.h"
+#include "llvm/ADT/TypeSwitch.h"
 #include "mlir/IR/DialectImplementation.h"
 #include "mlir/IR/OpDefinition.h"
 #include "mlir/Interfaces/FoldInterfaces.h"