diff --git a/sycl-fusion/jit-compiler/include/JITContext.h b/sycl-fusion/jit-compiler/include/JITContext.h
index a947ee619c648..0b7e39f01a239 100644
--- a/sycl-fusion/jit-compiler/include/JITContext.h
+++ b/sycl-fusion/jit-compiler/include/JITContext.h
@@ -9,7 +9,6 @@
 #ifndef SYCL_FUSION_JIT_COMPILER_JITCONTEXT_H
 #define SYCL_FUSION_JIT_COMPILER_JITCONTEXT_H
 
-#include "llvm/IR/LLVMContext.h"
 #include <mutex>
 #include <shared_mutex>
 #include <unordered_map>
@@ -17,6 +16,10 @@
 #include "Kernel.h"
 #include "Parameter.h"
 
+namespace llvm {
+class LLVMContext;
+} // namespace llvm
+
 namespace jit_compiler {
 
 ///
diff --git a/sycl-fusion/jit-compiler/include/KernelFusion.h b/sycl-fusion/jit-compiler/include/KernelFusion.h
index 06ded8b2560f2..d4607df2e5660 100644
--- a/sycl-fusion/jit-compiler/include/KernelFusion.h
+++ b/sycl-fusion/jit-compiler/include/KernelFusion.h
@@ -13,7 +13,9 @@
 #include "Kernel.h"
 #include "Options.h"
 #include "Parameter.h"
+#include <cassert>
 #include <string>
+#include <variant>
 #include <vector>
 
 namespace jit_compiler {
diff --git a/sycl-fusion/jit-compiler/lib/JITContext.cpp b/sycl-fusion/jit-compiler/lib/JITContext.cpp
index e1714f8f4e337..fba490e78670d 100644
--- a/sycl-fusion/jit-compiler/lib/JITContext.cpp
+++ b/sycl-fusion/jit-compiler/lib/JITContext.cpp
@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "JITContext.h"
+#include "llvm/IR/LLVMContext.h"
 
 using namespace jit_compiler;