diff --git a/nvim/.config/nvim/lua/lsp.lua b/nvim/.config/nvim/lua/lsp.lua
index 15cb272..b7e5d0d 100644
--- a/nvim/.config/nvim/lua/lsp.lua
+++ b/nvim/.config/nvim/lua/lsp.lua
@@ -208,7 +208,7 @@ local on_attach = function(client, bufnr)
       buffer   = bufnr,
       callback = function()
         -- Taken from https://github.com/neovim/nvim-lspconfig/wiki/Code-Actions
-        local context  = { diagnostics = vim.lsp.diagnostic.get_line_diagnostics() }
+        local context  = { diagnostics = vim.diagnostic.get(bufnr) }
         local params   = lsp_util.make_range_params()
         params.context = context
         vim.lsp.buf_request(0, 'textDocument/codeAction', params, function(err, result, ctx, config)