diff --git a/lua/tokyonight/theme.lua b/lua/tokyonight/theme.lua
index e2b8dab..067f650 100644
--- a/lua/tokyonight/theme.lua
+++ b/lua/tokyonight/theme.lua
@@ -194,6 +194,9 @@ function M.setup(config)
     -- TSBoolean           = { };    -- For booleans.
     -- TSCharacter         = { };    -- For characters.
     -- TSComment           = { };    -- For comment blocks.
+    TSNote = { fg = c.bg, bg = c.info },
+    TSWarning = { fg = c.bg, bg = c.warning },
+    TSDanger = { fg = c.bg, bg = c.error },
     TSConstructor = { fg = c.magenta }, -- For constructor calls and definitions: `= { }` in Lua, and Java constructors.
     -- TSConditional       = { };    -- For keywords related to conditionnals.
     -- TSConstant          = { };    -- For constants