diff --git a/lua/nvim-treesitter/highlight.lua b/lua/nvim-treesitter/highlight.lua
index 867c70caf..a5ce45337 100644
--- a/lua/nvim-treesitter/highlight.lua
+++ b/lua/nvim-treesitter/highlight.lua
@@ -28,6 +28,7 @@ hlmap["character"] = "TSCharacter"
 hlmap["number"] = "TSNumber"
 hlmap["boolean"] = "TSBoolean"
 hlmap["float"] = "TSFloat"
+hlmap["attribute"] = "TSAttribute"
 
 -- Functions
 hlmap["function"] = "TSFunction"
diff --git a/plugin/nvim-treesitter.vim b/plugin/nvim-treesitter.vim
index 90f997eb1..33db1aed2 100644
--- a/plugin/nvim-treesitter.vim
+++ b/plugin/nvim-treesitter.vim
@@ -44,6 +44,7 @@ highlight default link TSMethod Function
 highlight default link TSField Identifier
 highlight default link TSProperty Identifier
 highlight default link TSConstructor Special
+highlight default link TSAttribute Identifier
 
 highlight default link TSConditional Conditional
 highlight default link TSRepeat Repeat