diff --git a/lua/rose-pine/theme.lua b/lua/rose-pine/theme.lua
index d37eef1..0e30250 100644
--- a/lua/rose-pine/theme.lua
+++ b/lua/rose-pine/theme.lua
@@ -118,7 +118,7 @@ function M._load(options)
 	h('Tag', { fg = p.rose })
 	h('Todo', { fg = p.iris })
 	h('Type', { fg = p.foam })
-	h('Typedef', { fg = p.foam })
+	h('Typedef', { link = 'Type' })
 	h('Underlined', { underline = true })
 
 	h('htmlArg', { fg = p.iris })
@@ -202,37 +202,46 @@ function M._load(options)
 	h('healthWarning', { fg = groups.warn })
 
 	-- Treesitter
-	h('@annotation', { link = 'PreProc' })
-	h('@attribute', { link = 'PreProc' })
 	h('@boolean', { link = 'Boolean' })
 	h('@character', { link = 'Character' })
+	h('@character.special', { link = '@character' })
 	h('@class', { fg = p.foam })
 	h('@comment', { link = 'Comment' })
 	h('@conditional', { link = 'Conditional' })
-	h('@constant', { fg = p.foam })
+	h('@constant', { link = 'Constant' })
 	h('@constant.builtin', { fg = p.love })
+	h('@constant.macro', { link = '@constant' })
 	h('@constructor', { fg = p.foam })
 	h('@field', { fg = p.foam })
-	h('@function', { fg = p.rose })
+	h('@function', { link = 'Function' })
 	h('@function.builtin', { fg = p.love })
-	-- h('@defaultLibrary', {})
-	h('@include', { fg = p.pine })
-	h('@keyword', { fg = p.pine })
+	h('@function.macro', { link = '@function' })
+	h('@include', { link = 'Include' })
+	h('@interface', { fg = p.foam })
+	h('@keyword', { link = 'Keyword' })
 	h('@keyword.operator', { fg = p.subtle })
-	h('@label', { fg = p.foam })
-	h('@namespace', { link = 'Include' })
+	h('@label', { link = 'Label' })
+	h('@macro', { link = 'Macro' })
+	h('@method', { fg = p.iris })
+	-- h('@modifier', { link = '' })
+	-- h('@namespace', { link = '' })
 	h('@number', { link = 'Number' })
-	h('@operator', { fg = p.subtle })
+	h('@operator', { link = 'Operator' })
 	h('@parameter', { fg = p.iris, italic = maybe.italic })
+	h('@preproc', { link = 'PreProc' })
 	h('@property', { fg = p.foam, italic = maybe.italic })
-	h('@punctuation.bracket', { fg = groups.punctuation })
-	h('@punctuation.delimiter', { fg = groups.punctuation })
-	h('@punctuation.special', { fg = groups.punctuation })
+	h('@punctuation', { fg = groups.punctuation })
+	h('@punctuation.bracket', { link = '@punctuation' })
+	h('@punctuation.delimiter', { link = '@punctuation' })
+	h('@punctuation.special', { link = '@punctuation' })
+	h('@regexp', { link = 'String' })
+	h('@repeat', { link = 'Repeat' })
+	h('@storageclass', { link = 'StorageClass' })
 	h('@string', { link = 'String' })
 	h('@string.escape', { fg = p.pine })
 	h('@string.special', { link = '@string' })
 	h('@symbol', { link = 'Identifier' })
-	h('@tag', { fg = p.foam })
+	h('@tag', { link = 'Tag' })
 	h('@tag.attribute', { fg = p.iris })
 	h('@tag.delimiter', { fg = p.subtle })
 	h('@text', { fg = p.text })
@@ -243,24 +252,15 @@ function M._load(options)
 	h('@text.math', { link = 'Special' })
 	h('@text.environment', { link = 'Macro' })
 	h('@text.environment.name', { link = 'Type' })
-	h('@text.title', { fg = groups.headings.h1, bold = true })
+	h('@text.title', { link = 'Title' })
 	h('@text.uri', { fg = groups.link })
 	h('@text.note', { link = 'SpecialComment' })
-	h('@text.warning', { link = 'Todo' })
-	h('@text.danger', { link = 'WarningMsg' })
+	h('@text.warning', { fg = groups.warn })
+	h('@text.danger', { fg = groups.error })
 	h('@todo', { link = 'Todo' })
 	h('@type', { link = 'Type' })
 	h('@variable', { fg = p.text, italic = maybe.italic })
 	h('@variable.builtin', { fg = p.love })
-	-- h('@struct', {})
-	-- h('@enum', {})
-	-- h('@enumMember', {})
-	-- h('@event', {})
-	-- h('@interface', {})
-	-- h('@modifier', {})
-	-- h('@regexp', {})
-	-- h('@typeParameter', {})
-	-- h('@decorator', {})
 
 	-- vim.lsp.buf.document_highlight()
 	h('LspReferenceText', { bg = p.highlight_med })