diff --git a/plugins/by-name/gitsigns/settings-options.nix b/plugins/by-name/gitsigns/settings-options.nix
index 9cc707b6..8545ecb9 100644
--- a/plugins/by-name/gitsigns/settings-options.nix
+++ b/plugins/by-name/gitsigns/settings-options.nix
@@ -18,7 +18,7 @@ in
           return false
         end
         -- Setup keymaps
-        vim.api.nvim_buf_set_keymap(bufnr, 'n', 'hs', '<cmd>lua require"gitsigns".stage_hunk()<CR>', {})
+        vim.keymap.set('n', 'hs', '<cmd>lua require"gitsigns".stage_hunk()<CR>', { buffer = bufnr })
         ... -- More keymaps
       end
     ```
diff --git a/tests/test-sources/plugins/by-name/gitsigns/default.nix b/tests/test-sources/plugins/by-name/gitsigns/default.nix
index 5e637edc..33dbfb76 100644
--- a/tests/test-sources/plugins/by-name/gitsigns/default.nix
+++ b/tests/test-sources/plugins/by-name/gitsigns/default.nix
@@ -29,7 +29,7 @@
               return false
             end
             -- Setup keymaps
-            vim.api.nvim_buf_set_keymap(bufnr, 'n', 'hs', '<cmd>lua require"gitsigns".stage_hunk()<CR>', {})
+            vim.keymap.set('n', 'hs', '<cmd>lua require"gitsigns".stage_hunk()<CR>', { buffer = bufnr })
           end
         '';
         watch_gitdir = {