diff --git a/pkg/persistence/dm_trove_queries.go b/pkg/persistence/dm_trove_queries.go
index b85b595..ec1f029 100644
--- a/pkg/persistence/dm_trove_queries.go
+++ b/pkg/persistence/dm_trove_queries.go
@@ -121,7 +121,9 @@ func (p Profile) SaveDMTrove(trove DMTrove, should_download bool) {
 				url.IsContentDownloaded = true
 
 				// Update it in the DB
-				_, err = p.DB.NamedExec(`update chat_message_urls set is_content_downloaded = :is_content_downloaded where id = :id`, url)
+				_, err = p.DB.NamedExec(`
+					update chat_message_urls set is_content_downloaded = :is_content_downloaded where chat_message_id = :chat_message_id
+				`, url)
 				if err != nil {
 					panic(err)
 				}