diff --git a/lua/custom/plugins/undotree.lua b/lua/custom/plugins/undotree.lua
index c09143e7..0ba2ed64 100644
--- a/lua/custom/plugins/undotree.lua
+++ b/lua/custom/plugins/undotree.lua
@@ -3,4 +3,7 @@ return {
   keys = {
     { '<leader>u', '<cmd>UndotreeToggle<cr>', desc = 'Toggle undotree' },
   },
+  config = function()
+    vim.g.undotree_SetFocusWhenToggle = 1
+  end,
 }