diff --git a/MVMCoreUI/Atomic/Molecules/HorizontalCombinationViews/TwoLinkViewModel.swift b/MVMCoreUI/Atomic/Molecules/HorizontalCombinationViews/TwoLinkViewModel.swift index c9ae177b..893d2753 100644 --- a/MVMCoreUI/Atomic/Molecules/HorizontalCombinationViews/TwoLinkViewModel.swift +++ b/MVMCoreUI/Atomic/Molecules/HorizontalCombinationViews/TwoLinkViewModel.swift @@ -8,7 +8,7 @@ import Foundation -public class TwoLinkViewModel: MoleculeModelProtocol { +public class TwoLinkViewModel: ParentMoleculeModelProtocol { public static var identifier: String = "twoLinkView" public var id: String = UUID().uuidString @@ -16,6 +16,8 @@ public class TwoLinkViewModel: MoleculeModelProtocol { public var rightLink: LinkModel? public var leftLink: LinkModel? + public var children: [MoleculeModelProtocol] { [rightLink, leftLink].compactMap{ $0 } } + private enum CodingKeys: String, CodingKey { case id case moleculeName