diff --git a/lua/navigator/dochighlight.lua b/lua/navigator/dochighlight.lua
index 97b6ede..e16d0e4 100644
--- a/lua/navigator/dochighlight.lua
+++ b/lua/navigator/dochighlight.lua
@@ -209,7 +209,7 @@ _G.nav_doc_hl = function()
   local bufnr = vim.api.nvim_get_current_buf()
   local ref_params = vim.lsp.util.make_position_params()
   vim.lsp.for_each_buffer_client(bufnr, function(client, client_id, bufnr)
-    if client.server_capabilities.documentHighlightProvider then
+    if client.server_capabilities.documentHighlightProvider == true then
       client.request('textDocument/documentHighlight', ref_params, handle_document_highlight, bufnr)
     end
   end)
diff --git a/lua/navigator/lspclient/attach.lua b/lua/navigator/lspclient/attach.lua
index 3be7800..667a612 100644
--- a/lua/navigator/lspclient/attach.lua
+++ b/lua/navigator/lspclient/attach.lua
@@ -34,7 +34,7 @@ M.on_attach = function(client, bufnr)
     bufnr = bufnr,
   })
 
-  if client.server_capabilities.documentHighlightProvider then
+  if client.server_capabilities.documentHighlightProvider == true then
     require('navigator.dochighlight').documentHighlight()
   end