diff --git a/lua/rose-pine.lua b/lua/rose-pine.lua
index 0aa5c78..33b0d21 100644
--- a/lua/rose-pine.lua
+++ b/lua/rose-pine.lua
@@ -76,6 +76,9 @@ local function set_highlights()
 		IndentBlanklineChar = { fg = palette.muted, nocombine = true },
 		IndentBlanklineSpaceChar = { fg = palette.muted, nocombine = true },
 		IndentBlanklineSpaceCharBlankline = { fg = palette.muted, nocombine = true },
+
+		-- folke/which-key.nvim
+		WhichKeyFloat = { bg = groups.panel },
 	}
 	local default_highlights = {
 		ColorColumn = { bg = palette.surface },
@@ -519,10 +522,12 @@ local function set_highlights()
 
 		-- folke/which-key.nvim
 		WhichKey = { fg = palette.iris },
+		WhichKeyBorder = make_border(),
 		WhichKeyDesc = { fg = palette.gold },
-		WhichKeyFloat = { bg = groups.panel },
 		WhichKeyGroup = { fg = palette.foam },
+		WhichKeyNormal = { link = "NormalFloat" },
 		WhichKeySeparator = { fg = palette.subtle },
+		WhichKeyTitle = make_border(palette.foam),
 		WhichKeyValue = { fg = palette.rose },
 
 		-- lukas-reineke/indent-blankline.nvim
@@ -920,6 +925,7 @@ local function set_highlights()
 		TelescopeSelectionCaret = { fg = palette.rose },
 
 		WhichKeyFloat = { bg = "NONE" },
+		WhichKeyNormal = { bg = "NONE" },
 
 		IblIndent = { fg = palette.overlay, bg = "NONE" },
 		IblScope = { fg = palette.foam, bg = "NONE" },