diff --git a/browser/ai_chat/tools/content_agent_tool_base_test.cc b/browser/ai_chat/tools/content_agent_tool_base_test.cc
index 376641a71f7..66c1ffbbb4b 100644
--- a/browser/ai_chat/tools/content_agent_tool_base_test.cc
+++ b/browser/ai_chat/tools/content_agent_tool_base_test.cc
@@ -51,12 +51,12 @@ ContentAgentToolBaseTest::RunWithExpectedSuccess(
   base::RunLoop run_loop;
   optimization_guide::proto::Actions captured_actions;
   EXPECT_CALL(*mock_task_provider_, ExecuteActions)
-      .WillOnce(testing::Invoke([&captured_actions, &run_loop](
-                                    optimization_guide::proto::Actions actions,
-                                    Tool::UseToolCallback callback) {
+      .WillOnce([&captured_actions, &run_loop](
+                    optimization_guide::proto::Actions actions,
+                    Tool::UseToolCallback callback) {
         captured_actions = std::move(actions);
         run_loop.Quit();
-      }));
+      });
 
   tool_->UseTool(input_json, base::DoNothing());
   run_loop.Run();
diff --git a/components/ai_chat/core/browser/conversation_handler_unittest.cc b/components/ai_chat/core/browser/conversation_handler_unittest.cc
index a1d268f2f7a..75fcc7ad3fd 100644
--- a/components/ai_chat/core/browser/conversation_handler_unittest.cc
+++ b/components/ai_chat/core/browser/conversation_handler_unittest.cc
@@ -2575,8 +2575,7 @@ TEST_F(ConversationHandlerUnitTest,
   // Mock engine to return no new entry
   base::RunLoop run_loop;
   EXPECT_CALL(*engine, GenerateAssistantResponse)
-      .WillOnce(testing::DoAll(
-          // Complete the request
+      .WillOnce(  // Complete the request
           testing::WithArg<8>(
               [&](EngineConsumer::GenerationCompletedCallback callback) {
                 std::move(callback).Run(
@@ -2585,7 +2584,7 @@ TEST_F(ConversationHandlerUnitTest,
                             mojom::CompletionEvent::New("")),
                         std::nullopt)));
                 run_loop.QuitWhenIdle();
-              })));
+              }));
 
   // Submit a human entry to trigger the mocked response
   conversation_handler_->SubmitHumanConversationEntry("Test question",