diff --git a/lua/llm/config.lua b/lua/llm/config.lua
index 36d4e19..9b0f0ed 100644
--- a/lua/llm/config.lua
+++ b/lua/llm/config.lua
@@ -21,6 +21,7 @@ local default_config = {
     suffix = "<fim_suffix>",
   },
   debounce_ms = 150,
+  keymap_modes = {"i", "n"},
   accept_keymap = "<Tab>",
   dismiss_keymap = "<S-Tab>",
   tls_skip_verify_insecure = false,
diff --git a/lua/llm/keymaps.lua b/lua/llm/keymaps.lua
index 1356aed..490f611 100644
--- a/lua/llm/keymaps.lua
+++ b/lua/llm/keymaps.lua
@@ -27,16 +27,12 @@ function M.setup()
     return
   end
 
+  local keymap_modes = config.get().keymap_modes
   local accept_keymap = config.get().accept_keymap
   local dismiss_keymap = config.get().dismiss_keymap
 
-  vim.keymap.set("i", accept_keymap, accept_suggestion, { expr = true })
-
-  vim.keymap.set("n", accept_keymap, accept_suggestion, { expr = true })
-
-  vim.keymap.set("i", dismiss_keymap, dismiss_suggestion, { expr = true })
-
-  vim.keymap.set("n", dismiss_keymap, dismiss_suggestion, { expr = true })
+  vim.keymap.set(keymap_modes, accept_keymap, accept_suggestion, { expr = true })
+  vim.keymap.set(keymap_modes, dismiss_keymap, dismiss_suggestion, { expr = true })
 
   M.setup_done = true
 end