diff --git a/neovim/.config/nvim/after/plugin/undotree.lua b/neovim/.config/nvim/after/plugin/undotree.lua
index e63a278..cd31f6c 100644
--- a/neovim/.config/nvim/after/plugin/undotree.lua
+++ b/neovim/.config/nvim/after/plugin/undotree.lua
@@ -1,4 +1,7 @@
 vim.opt.undodir = os.getenv("HOME") .. "/.vim/undodir"
 vim.opt.undofile = true
 
-vim.keymap.set("n", "<leader>u", vim.cmd.UndotreeToggle)
+vim.keymap.set("n", "<leader>u", function()
+  vim.cmd.UndotreeToggle()
+  vim.cmd.UndotreeFocus()
+end)