diff --git a/modules/hm/vim.nix b/modules/hm/vim.nix
index c080a9b..eded771 100644
--- a/modules/hm/vim.nix
+++ b/modules/hm/vim.nix
@@ -53,6 +53,15 @@ let vp = pkgs.vimPlugins; in {
         config = builtins.readFile ./nvim/nvim-cmp-config.lua;
       }
 
+      # Undo tree
+      {
+        plugin = vp.undotree;
+        type = "lua";
+        config = ''
+          vim.keymap.set({ "n", "v" }, "<leader>u", vim.cmd.UndotreeToggle, { desc = "Toggle undo tree" })
+        '';
+      }
+
       # # Detect tabstop and shiftwidth automatically
       # vp.sleuth