diff --git a/MVMCoreUI/BaseClasses/TableViewCell.swift b/MVMCoreUI/BaseClasses/TableViewCell.swift
index e3ed8749..d90d0338 100644
--- a/MVMCoreUI/BaseClasses/TableViewCell.swift
+++ b/MVMCoreUI/BaseClasses/TableViewCell.swift
@@ -153,7 +153,7 @@ import UIKit
         // override the separator
         if let separator = model.line {
             addSeparatorsIfNeeded()
-            bottomSeparatorView?.set(with: separator, nil, nil)
+            bottomSeparatorView?.setStyle(separator.type)
         }
         
         if let moleculeModel = model as? MoleculeModelProtocol,