1 files changed, 6 insertions, 6 deletions
diff --git a/lua/pi/lsp/init.lua b/lua/pi/lsp/init.lua
index 31b5161..de6e95b 100644
--- a/lua/pi/lsp/init.lua
+++ b/lua/pi/lsp/init.lua
@@ -3,12 +3,12 @@ return {
     ops = vim.tbl_deep_extend("force", require "pi.lsp.default", ops)
 
     -- Configure LSP border for native LSP
-    vim.lsp.handlers["textDocument/hover"] = vim.lsp.with(vim.lsp.handlers.hover, {
-      border = "single",
-    })
-    vim.lsp.handlers["textDocument/signatureHelp"] = vim.lsp.with(vim.lsp.handlers.signature_help, {
-      border = "single",
-    })
+    local orig_open_floating_preview = vim.lsp.util.open_floating_preview
+    vim.lsp.util.open_floating_preview = function(contents, syntax, opts, ...)
+      opts = opts or {}
+      opts.border = opts.border or "single"
+      return orig_open_floating_preview(contents, syntax, opts, ...)
+    end
     
     local mason_servers = {}
     for server, options in pairs(ops.servers) do