diff --git a/after/ftplugin/json.lua b/after/ftplugin/json.lua
index 7f6f421..5e49adb 100644
--- a/after/ftplugin/json.lua
+++ b/after/ftplugin/json.lua
@@ -2,7 +2,7 @@
 --   Use treesitter for folding
 --   Disable the color column
 
-vim.opt.foldmethod="expr"
-vim.opt.foldexpr="nvim_treesitter#foldexpr()"
+vim.wo[0][0].foldexpr = 'v:lua.vim.treesitter.foldexpr()'
+vim.wo[0][0].foldmethod = 'expr'
 vim.opt.foldlevel=3
 vim.opt.colorcolumn=""
diff --git a/lua/plugins/treesitter.lua b/lua/plugins/treesitter.lua
index 1a2114b..29b8878 100644
--- a/lua/plugins/treesitter.lua
+++ b/lua/plugins/treesitter.lua
@@ -1,19 +1,5 @@
 -- Diagnostic keymaps
 
-vim.keymap.set(
-  'n',
-  '[d',
-  vim.diagnostic.goto_prev,
-  { desc = "[Diagnostic] Go to previous diagnostic message" }
-)
-
-vim.keymap.set(
-  'n',
-  ']d',
-  vim.diagnostic.goto_next,
-  { desc = "[Diagnostic] Go to next diagnostic message" }
-)
-
 vim.keymap.set(
   'n',
   '<leader>e',
diff --git a/lua/plugins/venv-selector.lua b/lua/plugins/venv-selector.lua
index c0ac6d2..7499b07 100644
--- a/lua/plugins/venv-selector.lua
+++ b/lua/plugins/venv-selector.lua
@@ -3,7 +3,7 @@
 
 return {
   'linux-cultist/venv-selector.nvim',
-  branch = 'regexp',
+  -- branch = 'regexp',
   dependencies = {
     'neovim/nvim-lspconfig',
     'nvim-telescope/telescope.nvim',