diff --git a/pkg/agent/loop_test.go b/pkg/agent/loop_test.go
index 702046b3c..675b28ec5 100644
--- a/pkg/agent/loop_test.go
+++ b/pkg/agent/loop_test.go
@@ -507,9 +507,6 @@ func TestRunLLMIteration_MessageOnlyToolCallsStopAfterOneRound(t *testing.T) {
 		t.Fatal("No default agent found")
 	}
 
-	al.beginRoundForMessageTool(agent)
-	al.updateToolContexts(agent, "telegram", "chat-1")
-
 	_, iteration, err := al.runLLMIteration(
 		context.Background(),
 		agent,