diff --git a/.config/nvim/lua/plugins/undotree.lua b/.config/nvim/lua/plugins/undotree.lua
index f87fdc0..fea3779 100644
--- a/.config/nvim/lua/plugins/undotree.lua
+++ b/.config/nvim/lua/plugins/undotree.lua
@@ -1,6 +1,12 @@
 return {
     "mbbill/undotree",
     keys = {
-        { "<C-u>", "<cmd>UndotreeToggle<cr><cmd>UndotreeFocus<cr>" },
+        {
+            "<C-u>",
+            function()
+                vim.cmd.UndotreeToggle();
+                vim.cmd.UndotreeFocus();
+            end
+        },
     },
 }