diff --git a/doom-modeline-segments.el b/doom-modeline-segments.el
index a67d615..8d99f19 100644
--- a/doom-modeline-segments.el
+++ b/doom-modeline-segments.el
@@ -122,8 +122,8 @@
 (declare-function symbol-overlay-get-list 'symbol-overlay)
 (declare-function symbol-overlay-get-symbol 'symbol-overlay)
 (declare-function symbol-overlay-rename 'symbol-overlay)
-(declare-function undo-tree-redo 'undo-tree)
-(declare-function undo-tree-undo 'undo-tree)
+(declare-function undo-tree-redo-1 'undo-tree)
+(declare-function undo-tree-undo-1 'undo-tree)
 (declare-function window-numbering-clear-mode-line 'window-numbering)
 (declare-function window-numbering-get-number-string 'window-numbering)
 (declare-function window-numbering-install-mode-line 'window-numbering)
@@ -221,8 +221,8 @@ Uses `all-the-icons-material' to fetch the icon."
 (add-hook 'after-change-functions #'doom-modeline-update-buffer-file-state-icon)
 (add-hook 'clone-indirect-buffer-hook #'doom-modeline-update-buffer-file-state-icon)
 (advice-add #'undo :after #'doom-modeline-update-buffer-file-state-icon)
-(advice-add #'undo-tree-undo :after #'doom-modeline-update-buffer-file-state-icon)
-(advice-add #'undo-tree-redo :after #'doom-modeline-update-buffer-file-state-icon)
+(advice-add #'undo-tree-undo-1 :after #'doom-modeline-update-buffer-file-state-icon)
+(advice-add #'undo-tree-redo-1 :after #'doom-modeline-update-buffer-file-state-icon)
 (advice-add #'narrow-to-region :after #'doom-modeline-update-buffer-file-state-icon)
 (advice-add #'widen :after #'doom-modeline-update-buffer-file-state-icon)
 
@@ -267,8 +267,8 @@ mouse-1: Previous buffer\nmouse-3: Next buffer"
 (advice-add #'rename-buffer :after #'doom-modeline-update-buffer-file-name)
 (advice-add #'set-visited-file-name :after #'doom-modeline-update-buffer-file-name)
 (advice-add #'undo :after #'doom-modeline-update-buffer-file-name)
-(advice-add #'undo-tree-undo :after #'doom-modeline-update-buffer-file-name)
-(advice-add #'undo-tree-redo :after #'doom-modeline-update-buffer-file-name)
+(advice-add #'undo-tree-undo-1 :after #'doom-modeline-update-buffer-file-name)
+(advice-add #'undo-tree-redo-1 :after #'doom-modeline-update-buffer-file-name)
 (advice-add #'symbol-overlay-rename :after #'doom-modeline-update-buffer-file-name)
 
 (when (>= emacs-major-version 26)