diff --git a/tests/Feature/DailyReportServiceTest.php b/tests/Feature/DailyReportServiceTest.php
index 50106739..8e7ac0b5 100644
--- a/tests/Feature/DailyReportServiceTest.php
+++ b/tests/Feature/DailyReportServiceTest.php
@@ -26,7 +26,7 @@ public function test_notify(): void
 
         $user = User::factory()->create();
 
-        LlmDriverFacade::shouldReceive('driver->chat')
+        LlmDriverFacade::shouldReceive('driver->setToolType->chat')
             ->once()
             ->andReturn(
                 CompletionResponse::from([