diff --git a/after/plugin/treesj.lua b/after/plugin/treesj.lua
index ffb89c1..a23b3ce 100644
--- a/after/plugin/treesj.lua
+++ b/after/plugin/treesj.lua
@@ -4,7 +4,5 @@ tsj.setup {
   use_default_keymaps = false,
 }
 
-local nmap = require "opdavies.keymap".nmap
-
-nmap { "gJ", tsj.join }
-nmap { "gS", tsj.split }
+vim.keymap.set("n", "gJ", tsj.join)
+vim.keymap.set("n", "gS", tsj.split)