diff --git a/after/queries/lua/highlights.scm b/after/queries/lua/highlights.scm
index b0e4c15..110c988 100644
--- a/after/queries/lua/highlights.scm
+++ b/after/queries/lua/highlights.scm
@@ -1,2 +1,2 @@
 (field
-  name: (identifier) @TSFieldKey)
+  name: (identifier) @field.key)
diff --git a/lua/base46/init.lua b/lua/base46/init.lua
index 1c87f3f..8a24782 100644
--- a/lua/base46/init.lua
+++ b/lua/base46/init.lua
@@ -44,7 +44,7 @@ M.load_all_highlights = function()
   local clear_hl = require("base46").clear_highlights
 
   clear_hl "BufferLine"
-  clear_hl "TS"
+  clear_hl "@"
 
   reload "base46.integrations"
   reload "base46.chadlights"
diff --git a/lua/base46/integrations/treesitter.lua b/lua/base46/integrations/treesitter.lua
index 935b0bf..1e34599 100644
--- a/lua/base46/integrations/treesitter.lua
+++ b/lua/base46/integrations/treesitter.lua
@@ -81,7 +81,7 @@ return {
     fg = theme.base08,
   },
 
-  TSParameterReference = {
+  ["@reference"] = {
     fg = theme.base05,
   },
 
@@ -169,7 +169,7 @@ return {
     underline = true,
   },
 
-  TSCurrentScope = {
+  ["@scope"] = {
     bold = true,
   },
 
@@ -177,7 +177,7 @@ return {
     fg = theme.base08,
   },
 
-  TSFieldKey = {
+  ["@field.key"] = {
     fg = theme.base0D,
   },