diff --git a/Signal/src/ViewControllers/Photos/ImagePickerController.swift b/Signal/src/ViewControllers/Photos/ImagePickerController.swift
index 1b5a201af1..85e8493fd7 100644
--- a/Signal/src/ViewControllers/Photos/ImagePickerController.swift
+++ b/Signal/src/ViewControllers/Photos/ImagePickerController.swift
@@ -203,6 +203,8 @@ class ImagePickerGridController: UICollectionViewController, PhotoLibraryDelegat
             let attachmentPromise: Promise<SignalAttachment> = photoCollectionContents.outgoingAttachment(for: asset)
             delegate.imagePicker(self, didSelectAsset: asset, attachmentPromise: attachmentPromise)
             collectionView.selectItem(at: indexPath, animated: true, scrollPosition: [])
+            updateDoneButtonAppearance()
+
         case .deselect:
             guard isSelected(indexPath: indexPath) else {
                 return
@@ -210,6 +212,7 @@ class ImagePickerGridController: UICollectionViewController, PhotoLibraryDelegat
 
             delegate.imagePicker(self, didDeselectAsset: asset)
             collectionView.deselectItem(at: indexPath, animated: true)
+            updateDoneButtonAppearance()
         }
     }