diff --git a/lua/rose-pine/theme.lua b/lua/rose-pine/theme.lua
index 0e4c4c2..01138a4 100644
--- a/lua/rose-pine/theme.lua
+++ b/lua/rose-pine/theme.lua
@@ -176,7 +176,9 @@ local theme = {
 
 	DiagnosticHint = { fg = config.colors.hint },
 	DiagnosticInfo = { fg = config.colors.info },
+	DiagnosticInformation = { link = "DiagnosticInfo" },
 	DiagnosticWarn = { fg = config.colors.warn },
+    DiagnosticWarning = { link = "DiagnosticWarn" },
 	DiagnosticError = { fg = config.colors.error },
 	DiagnosticDefaultHint = { fg = config.colors.hint },
 	DiagnosticDefaultInfo = { fg = config.colors.info },