diff --git a/init.lua b/init.lua
index 54aeae0..811cdf0 100644
--- a/init.lua
+++ b/init.lua
@@ -1,7 +1,6 @@
 require("config.lazy")
 
 vim.opt.mouse = ''
-
 vim.opt.number = true
 vim.opt.relativenumber = true
 vim.opt.rnu = true
@@ -61,6 +60,8 @@ vim.keymap.set("n", "<C-s>", function() ui.nav_file(4) end)
 
 -- UndoTree
 vim.keymap.set("n", "<leader>u", vim.cmd.UndotreeToggle)
+vim.opt.undodir = os.getenv("HOME") .. "/.vim/undodir"
+vim.opt.undofile = true
 
 -- Reserve a space in the gutter
 vim.opt.signcolumn = 'yes'