1 files changed, 2 insertions, 2 deletions
diff --git a/init.lua b/init.lua
index 8002e62..6f46895 100644
--- a/init.lua
+++ b/init.lua
@@ -594,7 +594,7 @@ end
 local function lspc_highlight_server_diagnostics(win, server_diagnostics, style)
   if not style then
     if not lspc.diagnostic_style_id then
-      if vis.API and vis.API > 3215 then
+      if (vis.API and vis.API > 3215) or (vis.ui and vis.ui.style_push) then
         lspc.diagnostic_style_id = vis.ui:style_push()
       else
         lspc.diagnostic_style_id = win.STYLE_LEXER_MAX
@@ -612,7 +612,7 @@ local function lspc_highlight_server_diagnostics(win, server_diagnostics, style)
 
   for _, diagnostic in ipairs(server_diagnostics) do
     local diagnostic_style = level_mapping[diagnostic.severity] or level_mapping[1]
-    assert(util.style_define(style, diagnostic_style))
+    assert(util.style_define(style, diagnostic_style, win))
 
     if lspc.highlight_diagnostics == 'range' then
       local range = diagnostic.vis_range