diff --git a/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift b/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift index 54ce4dfa40..c88ee2e026 100644 --- a/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift +++ b/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift @@ -38,6 +38,8 @@ public func configureLegacyAssetPicker(_ controller: TGMediaAssetsController, co controller.reminder = peer.id == context.account.peerId if let channel = peer as? TelegramChannel, case .broadcast = channel.info { controller.hasCoverButton = true + } else if peer.id == context.account.peerId { + controller.hasCoverButton = true } controller.presentScheduleController = { media, done in presentSchedulePicker(media, { time in diff --git a/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift b/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift index fc022bc9cb..507e1d781f 100644 --- a/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift +++ b/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift @@ -107,6 +107,8 @@ func presentLegacyMediaPickerGallery(context: AccountContext, peer: EnginePeer?, var hasCoverButton = false if case let .channel(channel) = peer, case .broadcast = channel.info { hasCoverButton = true + } else if peer?.id == context.account.peerId { + hasCoverButton = true } let legacyController = LegacyController(presentation: .custom, theme: presentationData.theme, initialLayout: nil)