diff --git a/lua/toggle-checkbox.lua b/lua/toggle-checkbox.lua
index 03714a7..80c1bcd 100644
--- a/lua/toggle-checkbox.lua
+++ b/lua/toggle-checkbox.lua
@@ -22,7 +22,8 @@ local M = {}
 M.toggle = function()
 	local bufnr = vim.api.nvim_buf_get_number(0)
 	local cursor = vim.api.nvim_win_get_cursor(0)
-	local current_line = vim.api.nvim_buf_get_lines(bufnr, cursor[1] - 1, cursor[1], false)[1] or ""
+	local start_line = cursor[1] - 1
+	local current_line = vim.api.nvim_buf_get_lines(bufnr, start_line, start_line + 1, false)[1] or ""
 
 	-- If the line contains a checked checkbox then uncheck it.
 	-- Otherwise, if it contains an unchecked checkbox, check it.
@@ -33,7 +34,7 @@ M.toggle = function()
 		new_line = checkbox.uncheck(current_line)
 	end
 
-	vim.api.nvim_buf_set_lines(bufnr, cursor[1] - 1, cursor[1], false, { new_line })
+	vim.api.nvim_buf_set_lines(bufnr, start_line, start_line + 1, false, { new_line })
 	vim.api.nvim_win_set_cursor(0, cursor)
 end