diff --git a/home/jonas/static/doom-config/config.el b/home/jonas/static/doom-config/config.el
index bcbf70f..ceb58cc 100644
--- a/home/jonas/static/doom-config/config.el
+++ b/home/jonas/static/doom-config/config.el
@@ -101,8 +101,8 @@
               ("C-TAB" . 'copilot-accept-completion-by-word)
               ("C-<tab>" . 'copilot-accept-completion-by-word)))
 
+(add-hook! 'prog-mode-hook #'undo-tree-mode)
 (after! undo-tree
-  (add-hook! 'prog-mode-hook #'undo-tree-mode)
   (setq undo-tree-auto-save-history t)
   (setq undo-tree-enable-undo-in-region nil)
   (setq undo-tree-history-directory-alist '(("." . "~/.cache/doom/undo"))))