diff --git a/tests/Feature/Models/MessageTest.php b/tests/Feature/Models/MessageTest.php
index 8415a09d..abe176f0 100644
--- a/tests/Feature/Models/MessageTest.php
+++ b/tests/Feature/Models/MessageTest.php
@@ -85,7 +85,7 @@ public function test_run(): void
 
         $message = Message::factory()->user()->create([
             'meta_data' => MetaDataDto::from([
-                'tool' => 'completion',
+                'tool' => 'chat',
             ]),
         ]);
 
@@ -115,14 +115,14 @@ public function test_rerun(): void
         $message = Message::factory()->user()->create([
             'chat_id' => $chat->id,
             'meta_data' => MetaDataDto::from([
-                'tool' => 'completion',
+                'tool' => 'chat',
             ]),
         ]);
 
         $messageAssistant = Message::factory()->assistant()->create([
             'chat_id' => $chat->id,
             'meta_data' => MetaDataDto::from([
-                'tool' => 'completion',
+                'tool' => 'chat',
             ]),
         ]);