diff --git a/app/src/main/java/org/thoughtcrime/securesms/database/AttachmentTable.kt b/app/src/main/java/org/thoughtcrime/securesms/database/AttachmentTable.kt
index 0e4df2be03..a214119cbf 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/database/AttachmentTable.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/database/AttachmentTable.kt
@@ -1677,6 +1677,10 @@ class AttachmentTable(
         values.put(DATA_RANDOM, fileWriteResult.random)
         values.put(DATA_HASH_START, fileWriteResult.hash)
         values.put(DATA_HASH_END, fileWriteResult.hash)
+
+        if (archiveRestore) {
+          values.put(ARCHIVE_TRANSFER_STATE, ArchiveTransferState.FINISHED.value)
+        }
       }
 
       val visualHashString = existingPlaceholder.getVisualHashStringOrNull()