WWK

I've had the same issue and here is a quick fix. CGPoint offset = treeView.contentOffset; // Do collapse/expand stuffs here treeView.contentOffset = offset; So just set contentOffset to the original...

Any updates on this problem?

Just found a solution after some tackles. `viewWillLayoutSubviews()` does the job! ``` override func viewWillLayoutSubviews() { super.viewWillLayoutSubviews() moveToViewController(at: customIndexToMove) } ```

Definitely needs to add a support of sections.