diff --git a/ViperMCFSwift/Modules/SecondModule/Configurator/SecondModuleConfigurator.swift b/ViperMCFSwift/Modules/SecondModule/Configurator/SecondModuleConfigurator.swift index 5bcd251..f79b74d 100644 --- a/ViperMCFSwift/Modules/SecondModule/Configurator/SecondModuleConfigurator.swift +++ b/ViperMCFSwift/Modules/SecondModule/Configurator/SecondModuleConfigurator.swift @@ -20,6 +20,7 @@ class SecondModuleModuleConfigurator { private func configure(viewController: SecondModuleViewController) { let router = SecondModuleRouter() + router.transitionHandler = viewController let presenter = SecondModulePresenter() presenter.view = viewController diff --git a/ViperMCFSwift/Modules/SecondModule/Router/SecondModuleRouter.swift b/ViperMCFSwift/Modules/SecondModule/Router/SecondModuleRouter.swift index 40c0a0d..4b2c22d 100644 --- a/ViperMCFSwift/Modules/SecondModule/Router/SecondModuleRouter.swift +++ b/ViperMCFSwift/Modules/SecondModule/Router/SecondModuleRouter.swift @@ -9,5 +9,5 @@ import ViperMcFlurry class SecondModuleRouter: NSObject, SecondModuleRouterInput { - var transitionHandler: RamblerViperModuleTransitionHandlerProtocol! + weak var transitionHandler: RamblerViperModuleTransitionHandlerProtocol! } diff --git a/ViperMCFSwift/Modules/ThirdModule/Configurator/ThirdModuleConfigurator.swift b/ViperMCFSwift/Modules/ThirdModule/Configurator/ThirdModuleConfigurator.swift index a58fc2d..dfb281a 100644 --- a/ViperMCFSwift/Modules/ThirdModule/Configurator/ThirdModuleConfigurator.swift +++ b/ViperMCFSwift/Modules/ThirdModule/Configurator/ThirdModuleConfigurator.swift @@ -20,6 +20,7 @@ class ThirdModuleModuleConfigurator { private func configure(viewController: ThirdModuleViewController) { let router = ThirdModuleRouter() + router.transitionHandler = viewController let presenter = ThirdModulePresenter() presenter.view = viewController diff --git a/ViperMCFSwift/Modules/ThirdModule/Router/ThirdModuleRouter.swift b/ViperMCFSwift/Modules/ThirdModule/Router/ThirdModuleRouter.swift index 84c0a56..3957043 100644 --- a/ViperMCFSwift/Modules/ThirdModule/Router/ThirdModuleRouter.swift +++ b/ViperMCFSwift/Modules/ThirdModule/Router/ThirdModuleRouter.swift @@ -8,4 +8,6 @@ import ViperMcFlurry -class ThirdModuleRouter: ThirdModuleRouterInput { } +class ThirdModuleRouter: ThirdModuleRouterInput { + weak var transitionHandler: RamblerViperModuleTransitionHandlerProtocol! +}