diff --git a/nvim/init.lua b/nvim/init.lua
index 2bb87b7..5ab1ac2 100644
--- a/nvim/init.lua
+++ b/nvim/init.lua
@@ -178,18 +178,18 @@ end
 
 local function _pim_float(row, col, height, width, border)
     local buf = vim.api.nvim_create_buf(false, true)
-    vim.api.nvim_open_win(buf, true, {
-        relative    = "editor",
-        row         = row,
-        col         = col,
-        height      = math.max(1, height),
-        width       = math.max(1, width),
-        style       = "minimal",
-        border      = border or "none",
-        zindex      = 50,
-        winhighlight = "Normal:Normal,NormalNC:Normal",
+    local win = vim.api.nvim_open_win(buf, true, {
+        relative = "editor",
+        row      = row,
+        col      = col,
+        height   = math.max(1, height),
+        width    = math.max(1, width),
+        style    = "minimal",
+        border   = border or "none",
+        zindex   = 50,
     })
-    return vim.api.nvim_get_current_win()
+    vim.api.nvim_set_option_value("winhighlight", "Normal:Normal,NormalNC:Normal", { win = win })
+    return win
 end
 
 -- n/g/f: individual centered floating windows