diff --git a/tests/Feature/ProjectOrchestrateTest.php b/tests/Feature/ProjectOrchestrateTest.php
index dfb6c6ab..9450591b 100644
--- a/tests/Feature/ProjectOrchestrateTest.php
+++ b/tests/Feature/ProjectOrchestrateTest.php
@@ -31,13 +31,13 @@ public function test_tools(): void
         $this->assertDatabaseCount('messages', 0);
         $this->assertDatabaseCount('tasks', 0);
 
-        LlmDriverFacade::shouldReceive('driver->setToolType->chat')
+        LlmDriverFacade::shouldReceive('driver->setSystemPrompt->setToolType->chat')
             ->once()
             ->andReturn(
                 CompletionResponse::from($response)
             );
 
-        LlmDriverFacade::shouldReceive('driver->chat')
+        LlmDriverFacade::shouldReceive('driver->setToolType->setSystemPrompt->chat')
             ->once()
             ->andReturn(
                 CompletionResponse::from($response)