                     appIconImageView.removeFromSuperview()
                     appCollectionSnapshotView.removeFromSuperview()
                     wallpaperSnapshotView.removeFromSuperview()
+                    appCollectionContainerView.removeFromSuperview()
                     transitionContext.completeTransition(!transitionContext.transitionWasCancelled)
             })
         }

        let imageScale = springBoardView.window!.screen.scale * 2

        // Get a snapshot of the app collection without the selected button
        appIconButton.isHidden = true
        let snapshotImage = springBoardView.snapshotImage(withScale: imageScale, afterScreenUpdates: true)
        appIconButton.isHidden = false
        
        // Here we cheat a little bit by getting a snapshot of the dock using the full window
        // hierarchy (b/c it is a visual effect view, otherwise it doesn't retain the blur) and
        // placing it in the dock location of the original snapshot. The animation won't be perfect
        // but it's quick enough that the user won't notice this little cheat
        let dockSnapshotImage = viewController.dockView.fullWindowHierarchySnapshotImage(with: imageScale)!
        
        let imageBounds = springBoardView.bounds
        let finalImage = UIGraphicsImageRenderer(size: imageBounds.size, scale: imageScale, opaque: false).image { (context) in
            snapshotImage.draw(in: imageBounds)
            dockSnapshotImage.draw(in: viewController.dockView.frame)
        }

        return finalImage

                UIView.addKeyframe(withRelativeStartTime: 0.0, relativeDuration: 1.0, animations: {
                    appIconContainerView.frame = finalFrame
                    appCollectionSnapshotView.alpha = 0
                    appCollectionContainerView.transform = CGAffineTransform(scaleX: self.otherAppZoomScale, y: self.otherAppZoomScale)
                    wallpaperSnapshotView.transform = CGAffineTransform(scaleX: self.wallpaperZoomScale, y: self.wallpaperZoomScale)
                })
                }, completion: { _ in
