diff --git a/pkg/task/manager.go b/pkg/task/manager.go
index 3e98581e0a3..68092fefb56 100644
--- a/pkg/task/manager.go
+++ b/pkg/task/manager.go
@@ -16,8 +16,8 @@ type Manager[K comparable] struct {
 
 func (tm *Manager[K]) Submit(task *Task[K]) K {
 	if tm.updateID != nil {
+		tm.updateID(&tm.curID)
 		task.ID = tm.curID
-		tm.updateID(&task.ID)
 	}
 	tm.tasks.Store(task.ID, task)
 	tm.do(task)