diff --git a/include/libs/stream/tstream.h b/include/libs/stream/tstream.h
index e96997a6c3..5fe92570c9 100644
--- a/include/libs/stream/tstream.h
+++ b/include/libs/stream/tstream.h
@@ -656,7 +656,7 @@ typedef struct STaskStatusEntry {
   STaskId id;
   int32_t status;
   int32_t statusLastDuration; // to record the last duration of current status
-  int32_t stage;
+  int64_t stage;
   int32_t nodeId;
   int64_t verStart;         // start version in WAL, only valid for source task
   int64_t verEnd;           // end version in WAL, only valid for source task
diff --git a/source/dnode/mnode/impl/src/mndStream.c b/source/dnode/mnode/impl/src/mndStream.c
index 92f5b9bdca..1b1a14c92d 100644
--- a/source/dnode/mnode/impl/src/mndStream.c
+++ b/source/dnode/mnode/impl/src/mndStream.c
@@ -2713,7 +2713,7 @@ int32_t killActiveCheckpointTrans(SMnode *pMnode, const char* pDBName, size_t le
   return TSDB_CODE_SUCCESS;
 }
 
-static int32_t mndResetStatusFromCheckpoint(SMnode *pMnode) {
+static int32_t mndResetStatusFromCheckpoint(SMnode *pMnode, int64_t transId) {
   STrans* pTrans = mndAcquireTrans(pMnode, transId);
   if (pTrans != NULL) {
     mInfo("kill checkpoint transId:%d to reset task status", transId);