diff --git a/source/libs/stream/src/streamExec.c b/source/libs/stream/src/streamExec.c
index 4f942b951d02d196ee659a1b9b50ec5c56e97a15..16341cdbbcb6dedbd801a4929b80ba8ebdf26cc5 100644
--- a/source/libs/stream/src/streamExec.c
+++ b/source/libs/stream/src/streamExec.c
@@ -382,10 +382,11 @@ static int32_t streamTransferStateToStreamTask(SStreamTask* pTask) {
     return code;
   }
 
-  if (pTask->info.taskLevel == TASK_LEVEL__SOURCE) {
+  int32_t level = pTask->info.taskLevel;
+  if (level == TASK_LEVEL__SOURCE) {
     streamTaskFillHistoryFinished(pTask);
     streamTaskEndScanWAL(pTask);
-  } else { // do transfer task operator states.
+  } else if (level == TASK_LEVEL__AGG) { // do transfer task operator states.
     code = streamDoTransferStateToStreamTask(pTask);
     if (code != TSDB_CODE_SUCCESS) {  // todo handle this
       return code;