diff --git a/package.json b/package.json
index c3723cea..d9c00515 100644
--- a/package.json
+++ b/package.json
@@ -264,8 +264,12 @@
       "[yaml]": {
         "codeBlocks.npmPackageName": "@tree-sitter-grammars/tree-sitter-yaml"
       },
+      "[toml]": {
+        "codeBlocks.npmPackageName": "@tree-sitter-grammars/tree-sitter-toml"
+      },
       "[xml]": {
-        "codeBlocks.npmPackageName": "@tree-sitter-grammars/tree-sitter-xml"
+        "codeBlocks.npmPackageName": "@tree-sitter-grammars/tree-sitter-xml",
+        "codeBlocks.subdirectory": "xml"
       },
       "[markdown]": {
         "codeBlocks.npmPackageName": "@tree-sitter-grammars/tree-sitter-markdown"
diff --git a/src/test/suite/Installer.test.ts b/src/test/suite/Installer.test.ts
index c91dae6c..1a80c73b 100644
--- a/src/test/suite/Installer.test.ts
+++ b/src/test/suite/Installer.test.ts
@@ -52,11 +52,11 @@ suite("Installer integration tests", function () {
         ["CSS", "css", "body { color: red; }"],
         ["YAML", "yaml", "key: value"],
         ["JSON", "json", '{ "key": "value" }'],
-        // ["XML", "xml"],
+        ["XML", "xml"],
         ["Markdown", "markdown", "# Title"],
         // ["LaTeX", "latex"],
         ["Bash", "shellscript", "echo 'Hello, World!'"],
-        // ["TOML", "toml"],
+        ["TOML", "toml"],
         // ["Swift", "swift"],
         ["Kotlin", "kotlin", "fun main() { }"],
         ["Zig", "zig", 'const std = @import("std");\n\npub fn main() void { }'],