diff --git a/server/agent/agent.go b/server/agent/agent.go
index fba80b7..3c092d5 100644
--- a/server/agent/agent.go
+++ b/server/agent/agent.go
@@ -184,7 +184,7 @@
 		return fmt.Errorf("failed to generate subtasks for task: %w", err)
 	}
 
-	solutionURL, err2 := a.taskManager.ProposeSubTasks(ctx, task, analysis)
+	solutionURL, err2 := a.taskManager.ProposeSubTasks(ctx, task, analysis, a.Name)
 	if err2 != nil {
 		return fmt.Errorf("failed to propose subtasks for task: %w", err2)
 	}
diff --git a/server/agent/manager.go b/server/agent/manager.go
index ea7474d..db8f27c 100644
--- a/server/agent/manager.go
+++ b/server/agent/manager.go
@@ -167,5 +167,6 @@
 				slog.String("error", err.Error()))
 		}
 	}
-	return nil
+
+	return m.taskManager.Close()
 }