[fix-212] - Set up workflow pointer on both references - TT

wwt · Aug 28, 2022 · 2f547e0 · 2f547e0
1 parent e228160
commit 2f547e0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Sources/SwiftCurrent_SwiftUI/Views/ViewControllerWrapper.swift b/Sources/SwiftCurrent_SwiftUI/Views/ViewControllerWrapper.swift
@@ -18,7 +18,11 @@ public struct ViewControllerWrapper<F: FlowRepresentable & UIViewController>: Vi
     public typealias WorkflowInput = F.WorkflowInput
     public typealias WorkflowOutput = F.WorkflowOutput
 
-    public weak var _workflowPointer: AnyFlowRepresentable?
+    public weak var _workflowPointer: AnyFlowRepresentable? {
+        didSet {
+            vc._workflowPointer = _workflowPointer
+        }
+    }
 
     private var vc: F