diff --git a/src/kernels/bang/reduce_mean.cc b/src/kernels/bang/reduce_mean.cc
index 2dd77e10..1b55c2ca 100644
--- a/src/kernels/bang/reduce_mean.cc
+++ b/src/kernels/bang/reduce_mean.cc
@@ -1,6 +1,6 @@
-#include "operators/reduce_mean.h"
 #include "bang/bang_kernel_without_config.h"
 #include "bang/bang_runtime.h"
+#include "operators/reduce.h"
 
 namespace infini {
 class ReduceMeanCnnl : public BangKernelWithoutConfig {
diff --git a/src/kernels/kunlun/reduce_mean.cc b/src/kernels/kunlun/reduce_mean.cc
index 08a01fd6..c7cf19ac 100644
--- a/src/kernels/kunlun/reduce_mean.cc
+++ b/src/kernels/kunlun/reduce_mean.cc
@@ -1,6 +1,6 @@
-#include "operators/reduce_mean.h"
 #include "kunlun/kunlun_kernel_without_config.h"
 #include "kunlun/kunlun_runtime.h"
+#include "operators/reduce.h"
 
 namespace infini {
 class ReduceMeanXdnn : public KUNLUNKernelWithoutConfig {