diff --git a/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.cpp b/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.cpp
index 8446b089cc1..40fa10f8516 100644
--- a/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.cpp
+++ b/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.cpp
@@ -12,32 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-#include <Common/CPUAffinityManager.h>
-#include <Common/FailPoint.h>
-#include <Common/ThreadFactory.h>
-#include <Common/ThreadManager.h>
 #include <Common/TiFlashMetrics.h>
-#include <DataStreams/IProfilingBlockInputStream.h>
-#include <DataStreams/SquashingBlockOutputStream.h>
-#include <Flash/Coprocessor/DAGCodec.h>
-#include <Flash/Coprocessor/DAGUtils.h>
-#include <Flash/Mpp/GRPCReceiverContext.h>
-#include <Flash/Mpp/MPPTask.h>
+#include <Flash/Mpp/MPPTaskManager.h>
 #include <Flash/Mpp/MPPTaskScheduleEntry.h>
-#include <Flash/Mpp/MPPTunnelSet.h>
-#include <Flash/Mpp/Utils.h>
-#include <Flash/Statistics/traverseExecutors.h>
-#include <Flash/executeQuery.h>
-#include <Interpreters/ProcessList.h>
-#include <Storages/Transaction/KVStore.h>
-#include <Storages/Transaction/TMTContext.h>
 #include <fmt/core.h>
 
-#include <chrono>
-#include <ext/scope_guard.h>
-#include <magic_enum.hpp>
-#include <map>
-
 namespace DB
 {
 MPPTaskScheduleEntry::MPPTaskScheduleEntry(MPPTaskManager * manager_, const MPPTaskId & id_)
diff --git a/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.h b/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.h
index b23cf820974..60ccb9297c0 100644
--- a/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.h
+++ b/dbms/src/Flash/Mpp/MPPTaskScheduleEntry.h
@@ -14,26 +14,8 @@
 
 #pragma once
 
-#include <Common/Exception.h>
 #include <Common/Logger.h>
-#include <Common/MemoryTracker.h>
-#include <DataStreams/BlockIO.h>
-#include <Flash/Coprocessor/DAGContext.h>
-#include <Flash/Mpp/MPPReceiverSet.h>
 #include <Flash/Mpp/MPPTaskId.h>
-#include <Flash/Mpp/MPPTaskStatistics.h>
-#include <Flash/Mpp/MPPTunnel.h>
-#include <Flash/Mpp/MPPTunnelSet.h>
-#include <Flash/Mpp/TaskStatus.h>
-#include <Interpreters/Context.h>
-#include <common/logger_useful.h>
-#include <common/types.h>
-#include <kvproto/mpp.pb.h>
-
-#include <atomic>
-#include <boost/noncopyable.hpp>
-#include <memory>
-#include <unordered_map>
 
 namespace DB
 {