diff --git a/modules/plugins/assistant/codecompanion/config.nix b/modules/plugins/assistant/codecompanion/config.nix
index fa863b74..08fc8cb2 100644
--- a/modules/plugins/assistant/codecompanion/config.nix
+++ b/modules/plugins/assistant/codecompanion/config.nix
@@ -21,17 +21,7 @@ in {
         };
       };
 
-      treesitter = {
-        enable = true;
-
-        # Codecompanion depends on the YAML grammar being added. Below is
-        # an easy way of adding an user-configurable grammar package exposed
-        # by the YAML language module *without* enabling the whole YAML language
-        # module. The package is defined even when the module is disabled.
-        grammars = [
-          config.vim.languages.yaml.treesitter.package
-        ];
-      };
+      treesitter.enable = true;
 
       autocomplete.nvim-cmp = {
         sources = {codecompanion-nvim = "[codecompanion]";};