diff --git a/Display/ContainedViewLayoutTransition.swift b/Display/ContainedViewLayoutTransition.swift
index 7d2d8d9ee4..b277b8df4b 100644
--- a/Display/ContainedViewLayoutTransition.swift
+++ b/Display/ContainedViewLayoutTransition.swift
@@ -227,9 +227,7 @@ public extension ContainedViewLayoutTransition {
                 case .spring:
                     timingFunction = kCAMediaTimingFunctionSpring
                 }
-                node.layer.animateBoundsOriginXAdditive(from: offset, to: 0.0, duration: duration, timingFunction: timingFunction, completion: { completed in
-                    print("completed \(completed)")
-                })
+                node.layer.animateBoundsOriginXAdditive(from: offset, to: 0.0, duration: duration, timingFunction: timingFunction)
         }
     }
     
diff --git a/Display/ViewController.swift b/Display/ViewController.swift
index 2f11d1b6fe..ba369b2a78 100644
--- a/Display/ViewController.swift
+++ b/Display/ViewController.swift
@@ -384,6 +384,18 @@ open class ViewControllerPresentationArguments {
         }
     }
     
+    @available(iOSApplicationExtension 9.0, *)
+    public func registerForPreviewingNonNative(with delegate: UIViewControllerPreviewingDelegate, sourceView: UIView, theme: PeekControllerTheme) {
+        if self.traitCollection.forceTouchCapability != .available {
+            if self.previewingContext == nil {
+                let previewingContext = SimulatedViewControllerPreviewing(theme: theme, delegate: delegate, sourceView: sourceView, node: self.displayNode, present: { [weak self] c, a in
+                    self?.presentInGlobalOverlay(c, with: a)
+                })
+                self.previewingContext = previewingContext
+            }
+        }
+    }
+    
     @available(iOSApplicationExtension 9.0, *)
     open override func unregisterForPreviewing(withContext previewing: UIViewControllerPreviewing) {
         if self.previewingContext != nil {