diff --git a/submodules/Display/Source/ContainedViewLayoutTransition.swift b/submodules/Display/Source/ContainedViewLayoutTransition.swift
index ceb65eb6fd..8ec45f3bb9 100644
--- a/submodules/Display/Source/ContainedViewLayoutTransition.swift
+++ b/submodules/Display/Source/ContainedViewLayoutTransition.swift
@@ -408,7 +408,7 @@ public extension ContainedViewLayoutTransition {
         }
     }
     
-    func updateBounds(layer: CALayer, bounds: CGRect, force: Bool = false, completion: ((Bool) -> Void)? = nil) {
+    func updateBounds(layer: CALayer, bounds: CGRect, beginWithCurrentState: Bool = false, force: Bool = false, completion: ((Bool) -> Void)? = nil) {
         if layer.bounds.equalTo(bounds) && !force {
             completion?(true)
         } else {
@@ -420,7 +420,12 @@ public extension ContainedViewLayoutTransition {
                     completion(true)
                 }
             case let .animated(duration, curve):
-                let previousBounds = layer.bounds
+                let previousBounds: CGRect
+                if beginWithCurrentState, layer.animation(forKey: "position") != nil, let presentation = layer.presentation() {
+                    previousBounds = presentation.bounds
+                } else {
+                    previousBounds = layer.bounds
+                }
                 layer.bounds = bounds
                 layer.animateBounds(from: previousBounds, to: bounds, duration: duration, timingFunction: curve.timingFunction, mediaTimingFunction: curve.mediaTimingFunction, force: force, completion: { result in
                     if let completion = completion {
diff --git a/submodules/Display/Source/ListViewProtocol.swift b/submodules/Display/Source/ListViewProtocol.swift
index 51a838d7e4..dbfac0920a 100644
--- a/submodules/Display/Source/ListViewProtocol.swift
+++ b/submodules/Display/Source/ListViewProtocol.swift
@@ -67,6 +67,7 @@ public protocol ListView: ASDisplayNode {
         additionalScrollDistance: CGFloat,
         updateSizeAndInsets: ListViewUpdateSizeAndInsets?,
         stationaryItemRange: (Int, Int)?,
+        customAnimationTransition: ControlledTransition?,
         updateOpaqueState: Any?,
         completion: @escaping (ListViewDisplayedItemRange) -> Void
     )
@@ -102,6 +103,7 @@ public extension ListView {
         additionalScrollDistance: CGFloat = 0.0,
         updateSizeAndInsets: ListViewUpdateSizeAndInsets? = nil,
         stationaryItemRange: (Int, Int)? = nil,
+        customAnimationTransition: ControlledTransition? = nil,
         updateOpaqueState: Any?,
         completion: @escaping (ListViewDisplayedItemRange) -> Void = { _ in }
     ) {
@@ -114,6 +116,7 @@ public extension ListView {
             additionalScrollDistance: additionalScrollDistance,
             updateSizeAndInsets: updateSizeAndInsets,
             stationaryItemRange: stationaryItemRange,
+            customAnimationTransition: customAnimationTransition,
             updateOpaqueState: updateOpaqueState,
             completion: completion
         )