diff --git a/lib/database/drift_db.dart b/lib/database/drift_db.dart
index 5963525..bcb2206 100644
--- a/lib/database/drift_db.dart
+++ b/lib/database/drift_db.dart
@@ -49,8 +49,7 @@ class AppDatabase extends _$AppDatabase {
   }
 
   Future<int> updateMessage(ChatMessagesCompanion message) {
-    return (update(chatMessages)
-      ..where((m) => m.id.equals(message.id.value))).write(message);
+    return into(chatMessages).insert(message, mode: InsertMode.insertOrReplace);
   }
 
   Future<int> updateMessageStatus(String id, MessageStatus status) {