diff --git a/MVMCoreUI/Atomic/Extensions/UIToolbar+Extension.swift b/MVMCoreUI/Atomic/Extensions/UIToolbar+Extension.swift index 49826504..cad763a6 100644 --- a/MVMCoreUI/Atomic/Extensions/UIToolbar+Extension.swift +++ b/MVMCoreUI/Atomic/Extensions/UIToolbar+Extension.swift @@ -40,10 +40,7 @@ public extension UIToolbar { class func addDismissToolbar(to object: TextFieldOrView, delegate: Any?, action: Selector) { - let toolbar = createEmptyToolbar() - let space = UIBarButtonItem(barButtonSystemItem: .flexibleSpace, target: nil, action: nil) - let dismissButton = UIBarButtonItem(barButtonSystemItem: .done, target: delegate, action: action) - toolbar.items = [space, dismissButton] + let toolbar = Self.getToolbarWithDoneButton(delegate: delegate, action: action) switch object { case is UITextField: