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