From 2cab6e6d832a8e973bfde857ee16af4330302b32 Mon Sep 17 00:00:00 2001
From: Valentin GHIRARDI <valentin.ghirardi@gmail.com>
Date: Fri, 21 Nov 2025 14:07:10 +0100
Subject: [PATCH] refactor

---
 lua/config.lua | 4 ++--
 lua/vim.lua    | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/lua/config.lua b/lua/config.lua
index 689982f..542f9bf 100644
--- a/lua/config.lua
+++ b/lua/config.lua
@@ -17,10 +17,10 @@ return {
 	},
 
 	mappings = {
-	    -- Undo last edit (insert)
+	    -- Undo last edit (normal/visual/insert)
 	    undo = "<C-z>",
 
-	    -- Redo last edit (insert)
+	    -- Redo last edit (normal/visual/insert)
 	    redo = "<C-y>"
 	}
     },
diff --git a/lua/vim.lua b/lua/vim.lua
index 2964643..6c08ce0 100644
--- a/lua/vim.lua
+++ b/lua/vim.lua
@@ -20,13 +20,13 @@ vim.opt.clipboard = "unnamedplus"
 
 local mappings = config.mappings
 
-vim.keymap.set("i", mappings.undo, "<C-o>:undo<CR>", {
+vim.keymap.set({ "n", "v", "i" }, mappings.undo, "<C-o>:undo<CR>", {
     desc = "Undo last edit",
     noremap = true,
     silent = true
 })
 
-vim.keymap.set("i", mappings.redo, "<C-o>:redo<CR>", {
+vim.keymap.set({ "n", "v", "i" }, mappings.redo, "<C-o>:redo<CR>", {
     desc = "Redo last edit",
     noremap = true,
     silent = true