diff --git a/submodules/Display/Display/GlobalOverlayPresentationContext.swift b/submodules/Display/Display/GlobalOverlayPresentationContext.swift
index 2292200781..c30965ce97 100644
--- a/submodules/Display/Display/GlobalOverlayPresentationContext.swift
+++ b/submodules/Display/Display/GlobalOverlayPresentationContext.swift
@@ -23,10 +23,10 @@ func isViewVisibleInHierarchy(_ view: UIView, _ initial: Bool = true) -> Bool {
     }
 }
 
-private final class HierarchyTrackingNode: ASDisplayNode {
+public final class HierarchyTrackingNode: ASDisplayNode {
     private let f: (Bool) -> Void
     
-    init(_ f: @escaping (Bool) -> Void) {
+    public init(_ f: @escaping (Bool) -> Void) {
         self.f = f
         
         super.init()
@@ -34,13 +34,13 @@ private final class HierarchyTrackingNode: ASDisplayNode {
         self.isLayerBacked = true
     }
     
-    override func didEnterHierarchy() {
+    override public func didEnterHierarchy() {
         super.didEnterHierarchy()
         
         self.f(true)
     }
     
-    override func didExitHierarchy() {
+    override public func didExitHierarchy() {
         super.didExitHierarchy()
         
         self.f(false)