diff --git a/UIAnimatedTextField/Source/UIAnimatedTextField.swift b/UIAnimatedTextField/Source/UIAnimatedTextField.swift
index 0efdcbd..1d4562b 100644
--- a/UIAnimatedTextField/Source/UIAnimatedTextField.swift
+++ b/UIAnimatedTextField/Source/UIAnimatedTextField.swift
@@ -444,7 +444,7 @@ open class UIAnimatedTextField: UIView {
 
         let spacerItem = UIBarButtonItem(barButtonSystemItem: .flexibleSpace, target: nil, action: nil)
         let doneItem = UIBarButtonItem(title: doneTitle,
-                                       style: .plain,
+                                       style: .done,
                                        target: self,
                                        action: #selector(datePickerDoneAction))
         let attributes = [