diff --git a/lua/zenbones/specs/dark.lua b/lua/zenbones/specs/dark.lua
index 2e6e9a8..a8518aa 100644
--- a/lua/zenbones/specs/dark.lua
+++ b/lua/zenbones/specs/dark.lua
@@ -438,6 +438,17 @@ local function generate(p, opt)
 
 			FlashLabel                       { bg = p.water.lightness(p1.bg.l + 24), fg = p.fg },
 			FlashBackdrop                    { fg = Comment.fg },
+
+			NotifyERRORIcon                  { DiagnosticError },
+			NotifyERRORTitle                 { DiagnosticError },
+			NotifyWARNIcon                   { DiagnosticWarn },
+			NotifyWARNTitle                  { DiagnosticWarn },
+			NotifyINFOIcon                   { DiagnosticInfo },
+			NotifyINFOTitle                  { DiagnosticInfo },
+			NotifyDEBUGIcon                  { DiagnosticHint },
+			NotifyDEBUGTitle                 { DiagnosticHint },
+			NotifyTRACEIcon                  { DiagnosticHint },
+			NotifyTRACETitle                 { DiagnosticHint },
 		}
 	end)
 	-- stylua: ignore end
diff --git a/lua/zenbones/specs/light.lua b/lua/zenbones/specs/light.lua
index f22ea79..555e33b 100644
--- a/lua/zenbones/specs/light.lua
+++ b/lua/zenbones/specs/light.lua
@@ -438,6 +438,17 @@ local function generate(p, opt)
 
 			FlashLabel                       { bg = p.water.lightness(p1.bg.l - 15), fg = p.fg },
 			FlashBackdrop                    { fg = Comment.fg },
+
+			NotifyERRORIcon                  { DiagnosticError },
+			NotifyERRORTitle                 { DiagnosticError },
+			NotifyWARNIcon                   { DiagnosticWarn },
+			NotifyWARNTitle                  { DiagnosticWarn },
+			NotifyINFOIcon                   { DiagnosticInfo },
+			NotifyINFOTitle                  { DiagnosticInfo },
+			NotifyDEBUGIcon                  { DiagnosticHint },
+			NotifyDEBUGTitle                 { DiagnosticHint },
+			NotifyTRACEIcon                  { DiagnosticHint },
+			NotifyTRACETitle                 { DiagnosticHint },
 		}
 	end)
 	-- stylua: ignore end