diff --git a/lua/toggle-checkbox.lua b/lua/toggle-checkbox.lua index 7ad350f..8175c2c 100644 --- a/lua/toggle-checkbox.lua +++ b/lua/toggle-checkbox.lua @@ -16,11 +16,13 @@ local M = {} M.check = function() local new_line = current_line:gsub("%[ %]", checked_checkbox) vim.api.nvim_buf_set_lines(bufnr, cursor[1] - 1, cursor[1], false, { new_line }) + vim.api.nvim_win_set_cursor(0, cursor) end M.uncheck = function() local new_line = current_line:gsub("%[" .. checked_character .. "%]", unchecked_checkbox) vim.api.nvim_buf_set_lines(bufnr, cursor[1] - 1, cursor[1], false, { new_line }) + vim.api.nvim_win_set_cursor(0, cursor) end return M