diff --git a/lua/rose-pine/theme.lua b/lua/rose-pine/theme.lua
index b87b3ae..021c55d 100644
--- a/lua/rose-pine/theme.lua
+++ b/lua/rose-pine/theme.lua
@@ -281,55 +281,55 @@ function M.get()
     TSVariable = {fg = p.text, style = styles.italic},
     TSVariableBuiltin = {fg = p.love},
 
-		-- Treesitter
-		['@annotation'] = { link = 'PreProc' },
-		['@attribute'] = { link = 'PreProc' },
-		['@boolean'] = { link = 'Boolean' },
-		['@character'] = { link = 'Character' },
-		['@comment'] = { link = 'Comment' },
-		['@conditional'] = { link = 'Conditional' },
-		['@constant'] = { fg = p.foam },
-		['@constant.builtin'] = { fg = p.love },
-		['@constructor'] = { fg = p.foam },
-		['@field'] = { fg = p.foam },
-		['@function'] = { fg = p.rose },
-		['@function.builtin'] = { fg = p.love },
-		['@include'] = { fg = p.pine },
-		['@keyword'] = { fg = p.pine },
-		['@keyword.operator'] = { fg = p.subtle },
-		['@label'] = { fg = p.foam },
-		['@namespace'] = { link = 'Include' },
-		['@number'] = { link = 'Number' },
-		['@operator'] = { fg = p.subtle },
-		['@parameter'] = { fg = p.iris, style = styles.italic },
-		['@property'] = { fg = p.iris, style = styles.italic },
-		['@punctuation.bracket'] = { fg = groups.punctuation },
-		['@punctuation.delimiter'] = { fg = groups.punctuation },
-		['@punctuation.special'] = { fg = groups.punctuation },
-		['@string'] = { link = 'String' },
-		['@string.escape'] = { fg = p.pine },
-		['@string.special'] = { link = '@string' },
-		['@symbol'] = { link = 'Identifier' },
-		['@tag'] = { fg = p.foam },
-		['@tag.attribute'] = { link = '@property' },
-		['@tag.delimiter'] = { fg = p.subtle },
-		['@text'] = { fg = p.text },
-		['@text.strong'] = { bold = true },
-		['@text.emphasis'] = { italic = true },
-		['@text.underline'] = { underline = true },
-		['@text.strike'] = { strikethrough = true },
-		['@text.math'] = { link = 'Special' },
-		['@text.environment'] = { link = 'Macro' },
-		['@text.environment.name'] = { link = 'Type' },
-		['@text.title'] = { fg = groups.headings.h1, style = 'bold' },
-		['@text.uri'] = { fg = groups.link },
-		['@text.note'] = { link = 'SpecialComment' },
-		['@text.warning'] = { link = 'Todo' },
-		['@text.danger'] = { link = 'WarningMsg' },
-		['@todo'] = { link = 'Todo' },
-		['@type'] = { link = 'Type' },
-		['@variable'] = { fg = p.text, style = styles.italic },
-		['@variable.builtin'] = { fg = p.love },
+    -- Treesitter
+    ['@annotation'] = {link = 'PreProc'},
+    ['@attribute'] = {link = 'PreProc'},
+    ['@boolean'] = {link = 'TSBoolean'},
+    ['@character'] = {link = 'TSCharacter'},
+    ['@comment'] = {link = 'TSComment'},
+    ['@conditional'] = {link = 'Conditional'},
+    ['@constant'] = {link = 'TSConstant'},
+    ['@constant.builtin'] = {link = 'TSConstBuiltin'},
+    ['@constructor'] = {link = 'TSConstructor'},
+    ['@field'] = {link = 'TSField'},
+    ['@function'] = {link = 'TSFunction'},
+    ['@function.builtin'] = {link = 'TSFuncBuiltin'},
+    ['@include'] = {link = 'TSInclude'},
+    ['@keyword'] = {link = 'TSKeyword'},
+    ['@keyword.operator'] = {link = 'TSKeywordOperator'},
+    ['@label'] = {link = 'TSLabel'},
+    ['@namespace'] = {link = 'Include'},
+    ['@number'] = {link = 'TSNumber'},
+    ['@operator'] = {link = 'TSOperator'},
+    ['@parameter'] = {link = 'TSParameter'},
+    ['@property'] = {link = 'TSProperty'},
+    ['@punctuation.bracket'] = {fg = groups.punctuation},
+    ['@punctuation.delimiter'] = {fg = groups.punctuation},
+    ['@punctuation.special'] = {fg = groups.punctuation},
+    ['@string'] = {link = 'TSString'},
+    ['@string.escape'] = {link = 'TSStringEscape'},
+    ['@string.special'] = {link = 'TSStringSpecial'},
+    ['@symbol'] = {link = 'Identifier'},
+    ['@tag'] = {link = 'TSTag'},
+    ['@tag.attribute'] = {link = '@property'},
+    ['@tag.delimiter'] = {link = 'TSTagDelimiter'},
+    ['@text'] = {link = 'TSText'},
+    ['@text.strong'] = {bold = true},
+    ['@text.emphasis'] = {italic = true},
+    ['@text.underline'] = {underline = true},
+    ['@text.strike'] = {strikethrough = true},
+    ['@text.math'] = {link = 'Special'},
+    ['@text.environment'] = {link = 'Macro'},
+    ['@text.environment.name'] = {link = 'Type'},
+    ['@text.title'] = {link = 'TSTitle'},
+    ['@text.uri'] = {fg = groups.link},
+    ['@text.note'] = {link = 'SpecialComment'},
+    ['@text.warning'] = {link = 'Todo'},
+    ['@text.danger'] = {link = 'WarningMsg'},
+    ['@todo'] = {link = 'Todo'},
+    ['@type'] = {link = 'Type'},
+    ['@variable'] = {link = 'TSVariable'},
+    ['@variable.builtin'] = {link = 'TSVariableBuiltin'},
 
     -- vim.lsp.buf.document_highlight()
     LspReferenceText = {bg = p.highlight_med},