diff --git a/MVMCoreUI/BaseClasses/TableViewCell.swift b/MVMCoreUI/BaseClasses/TableViewCell.swift
index 053bedd0..e3ed8749 100644
--- a/MVMCoreUI/BaseClasses/TableViewCell.swift
+++ b/MVMCoreUI/BaseClasses/TableViewCell.swift
@@ -243,8 +243,8 @@ import UIKit
     open func setLines(with model: LineModel?, delegateObject: MVMCoreUIDelegateObject?, additionalData: [AnyHashable: Any]?, indexPath: IndexPath) {
         addSeparatorsIfNeeded()
         if let model = model {
-            topSeparatorView?.set(with: model, delegateObject, additionalData)
-            bottomSeparatorView?.set(with: model, delegateObject, additionalData)
+            topSeparatorView?.setStyle(model.type)
+            bottomSeparatorView?.setStyle(model.type)
         } else {
             topSeparatorView?.setStyle(.secondary)
             bottomSeparatorView?.setStyle(.secondary)