diff --git a/submodules/LegacyComponents/Sources/TGPhotoToolsController.m b/submodules/LegacyComponents/Sources/TGPhotoToolsController.m
index 1dfec01a9e..6956c9ab0f 100644
--- a/submodules/LegacyComponents/Sources/TGPhotoToolsController.m
+++ b/submodules/LegacyComponents/Sources/TGPhotoToolsController.m
@@ -499,7 +499,13 @@ const CGFloat TGPhotoEditorToolsLandscapePanelSize = TGPhotoEditorToolsPanelSize
     
     [TGPhotoEditorAnimation performBlock:^(__unused bool allFinished)
     {
-        [snapshotView removeFromSuperview];
+        if (self.intent == TGPhotoEditorControllerWallpaperIntent) {
+            TGDispatchAfter(0.3, dispatch_get_main_queue(), ^{
+                [snapshotView removeFromSuperview];
+            });
+        } else {
+            [snapshotView removeFromSuperview];
+        }
          
         if (completion != nil)
             completion();