diff --git a/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/Contents.json b/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/Contents.json
new file mode 100644
index 0000000..2b35ba7
--- /dev/null
+++ b/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/Contents.json
@@ -0,0 +1,22 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "scale" : "1x"
+    },
+    {
+      "filename" : "key_revoke@2x.png",
+      "idiom" : "universal",
+      "scale" : "2x"
+    },
+    {
+      "filename" : "key_revoke@3x.png",
+      "idiom" : "universal",
+      "scale" : "3x"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/key_revoke@2x.png b/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/key_revoke@2x.png
new file mode 100644
index 0000000..911009a
Binary files /dev/null and b/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/key_revoke@2x.png differ
diff --git a/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/key_revoke@3x.png b/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/key_revoke@3x.png
new file mode 100644
index 0000000..1313634
Binary files /dev/null and b/CustomKeyboard/KeyboardAssets.xcassets/key_revoke.imageset/key_revoke@3x.png differ
diff --git a/CustomKeyboard/View/KBToolBar.m b/CustomKeyboard/View/KBToolBar.m
index ba087c0..19ffe93 100644
--- a/CustomKeyboard/View/KBToolBar.m
+++ b/CustomKeyboard/View/KBToolBar.m
@@ -262,14 +262,15 @@ static const CGFloat kKBAIButtonHeight = 40;
 
 - (UIButton *)undoButtonInternal {
     if (!_undoButtonInternal) {
-        _undoButtonInternal = [UIButton buttonWithType:UIButtonTypeSystem];
-        _undoButtonInternal.layer.cornerRadius = 16;
-        _undoButtonInternal.layer.masksToBounds = YES;
-        _undoButtonInternal.backgroundColor = [UIColor colorWithWhite:1 alpha:0.9];
-        _undoButtonInternal.titleLabel.font = [UIFont systemFontOfSize:14 weight:UIFontWeightMedium];
-        [_undoButtonInternal setTitle:@"撤销删除" forState:UIControlStateNormal];
-        [_undoButtonInternal setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
-        _undoButtonInternal.contentEdgeInsets = UIEdgeInsetsMake(0, 10, 0, 10);
+        _undoButtonInternal = [UIButton buttonWithType:UIButtonTypeCustom];
+//        _undoButtonInternal.layer.cornerRadius = 16;
+//        _undoButtonInternal.layer.masksToBounds = YES;
+//        _undoButtonInternal.backgroundColor = [UIColor colorWithWhite:1 alpha:0.9];
+//        _undoButtonInternal.titleLabel.font = [UIFont systemFontOfSize:14 weight:UIFontWeightMedium];
+//        [_undoButtonInternal setTitle:@"撤销删除" forState:UIControlStateNormal];
+//        [_undoButtonInternal setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
+//        _undoButtonInternal.contentEdgeInsets = UIEdgeInsetsMake(0, 10, 0, 10);
+        [_undoButtonInternal setImage:[UIImage imageNamed:@"key_revoke"] forState:UIControlStateNormal];
         _undoButtonInternal.hidden = YES;
         _undoButtonInternal.alpha = 0.0;
         [_undoButtonInternal addTarget:self action:@selector(onUndo) forControlEvents:UIControlEventTouchUpInside];