diff --git a/src/plugin/ultrawork-db-model-override.ts b/src/plugin/ultrawork-db-model-override.ts
index 9009fb06..8a36609d 100644
--- a/src/plugin/ultrawork-db-model-override.ts
+++ b/src/plugin/ultrawork-db-model-override.ts
@@ -23,8 +23,8 @@ function tryUpdateMessageModel(
   if (result.changes === 0) return false
   if (variant) {
     db.prepare(
-      `UPDATE message SET data = json_set(data, '$.variant', ?) WHERE id = ?`,
-    ).run(variant, messageId)
+      `UPDATE message SET data = json_set(data, '$.variant', ?, '$.thinking', ?) WHERE id = ?`,
+    ).run(variant, variant, messageId)
   }
   return true
 }