diff --git a/lua/go/codelens.lua b/lua/go/codelens.lua
index 229461e..3698ab6 100644
--- a/lua/go/codelens.lua
+++ b/lua/go/codelens.lua
@@ -29,10 +29,10 @@ local M = {}
 
 function M.setup()
 
-  vim.cmd('highlight! link LspCodeLens WarningMsg')
-  vim.cmd('highlight! link LspCodeLensText WarningMsg')
-  vim.cmd('highlight! link LspCodeLensTextSign LspCodeLensText')
-  vim.cmd('highlight! link LspCodeLensTextSeparator Boolean')
+  vim.cmd('highlight! default link LspCodeLens WarningMsg')
+  vim.cmd('highlight! default link LspCodeLensText WarningMsg')
+  vim.cmd('highlight! default link LspCodeLensTextSign LspCodeLensText')
+  vim.cmd('highlight! default link LspCodeLensTextSeparator Boolean')
 
   vim.cmd('augroup go.codelenses')
   vim.cmd('  autocmd!')