diff --git a/home-manager/modules/programs/neovim.nix b/home-manager/modules/programs/neovim.nix
index 45cd344..7aa636c 100644
--- a/home-manager/modules/programs/neovim.nix
+++ b/home-manager/modules/programs/neovim.nix
@@ -31,7 +31,9 @@
       }
       vim-surround
       vim-commentary
-      nvim-treesitter.withAllGrammars
+      {
+        plugin = nvim-treesitter.withAllGrammars;
+      }
       nvim-nu
       null-ls-nvim
       vim-fugitive
@@ -39,6 +41,11 @@
     extraLuaConfig = ''
       vim.o.termguicolors = true
       require'nu'.setup{}
+      require'nvim-treesitter.configs'.setup {
+        highlight = {
+          enable = true,
+        },
+      }
     '';
     extraConfig = ''
       " Show relative line numbers
@@ -49,6 +56,8 @@
       set shiftwidth=4
       " Keep 4 lines off the edge of the screen when scrolling
       set scrolloff=4
+      " Djot setup
+      au BufNewFile,BufRead *.dj set filetype=djot
     '';
     withPython3 = true;
   };