diff --git a/lua/tokyonight/theme.lua b/lua/tokyonight/theme.lua
index fecd21d..aac3bac 100644
--- a/lua/tokyonight/theme.lua
+++ b/lua/tokyonight/theme.lua
@@ -680,6 +680,10 @@ function M.setup()
     NoiceCompletionItemKindEnumMember = { fg = c.green1, bg = c.none },
     NoiceCompletionItemKindOperator = { fg = c.green1, bg = c.none },
     NoiceCompletionItemKindSnippet = { fg = c.dark5, bg = c.none },
+
+    TreesitterContext = { bg = util.darken(c.fg_gutter, 0.8) },
+    Hlargs = { fg = c.yellow },
+    -- TreesitterContext = { bg = util.darken(c.bg_visual, 0.4) },
   }
 
   if not vim.diagnostic then