diff --git a/internal/datacoord/segment_allocation_policy.go b/internal/datacoord/segment_allocation_policy.go
index abcea9a69dfaaac399babff36019b5e14ac80d1a..9333e44b11e584c30fb56c203ab819f0bb05a404 100644
--- a/internal/datacoord/segment_allocation_policy.go
+++ b/internal/datacoord/segment_allocation_policy.go
@@ -185,7 +185,6 @@ const flushInterval = 2 * time.Second
 
 func flushPolicyV1(segment *SegmentInfo, t Timestamp) bool {
 	return segment.GetState() == commonpb.SegmentState_Sealed &&
-		segment.GetLastExpireTime() <= t &&
 		time.Since(segment.lastFlushTime) >= flushInterval &&
-		segment.currRows != 0
+		(segment.GetLastExpireTime() <= t && segment.currRows != 0 || (segment.IsImporting))
 }
diff --git a/internal/rootcoord/import_manager.go b/internal/rootcoord/import_manager.go
index a88176650da8902526c72941e7b15d0ef64ca839..c2b5c3a5a400bea94b6c1e8703e67a4fef7fbe85 100644
--- a/internal/rootcoord/import_manager.go
+++ b/internal/rootcoord/import_manager.go
@@ -731,7 +731,6 @@ func (m *importManager) getTaskState(tID int64) *milvuspb.GetImportStateResponse
 		},
 		Infos: make([]*commonpb.KeyValuePair, 0),
 	}
-	log.Debug("getting import task state", zap.Int64("task ID", tID))
 	// (1) Search in pending tasks list.
 	found := false
 	m.pendingLock.Lock()
@@ -771,6 +770,7 @@ func (m *importManager) getTaskState(tID int64) *milvuspb.GetImportStateResponse
 			zap.Error(err))
 	}
 	if found {
+		log.Info("getting import task state", zap.Int64("task ID", tID), zap.Any("state", resp.State), zap.Int64s("segment", resp.SegmentIds))
 		return resp
 	}
 	log.Debug("get import task state failed", zap.Int64("taskID", tID))