diff --git a/MVMCoreUI/Atomic/Molecules/Items/MoleculeTableViewCell.swift b/MVMCoreUI/Atomic/Molecules/Items/MoleculeTableViewCell.swift
index 4534a506..0c9dec3b 100644
--- a/MVMCoreUI/Atomic/Molecules/Items/MoleculeTableViewCell.swift
+++ b/MVMCoreUI/Atomic/Molecules/Items/MoleculeTableViewCell.swift
@@ -46,7 +46,7 @@ import UIKit
     }
     
     public override class func estimatedHeight(with model: MoleculeModelProtocol, _ delegateObject: MVMCoreUIDelegateObject?) -> CGFloat {
-        guard let moleculeModel = (model as? MoleculeContainerModel)?.molecule,
+        guard let moleculeModel = (model as? MoleculeListItemModel)?.molecule,
             let classType = MoleculeObjectMapping.shared()?.getMoleculeClass(moleculeModel),
             let height = classType.estimatedHeight(with: moleculeModel, delegateObject)
             else { return 80 }