diff --git a/lua/diagnostic/util.lua b/lua/diagnostic/util.lua
index cd6a7fb..92b9b91 100644
--- a/lua/diagnostic/util.lua
+++ b/lua/diagnostic/util.lua
@@ -134,7 +134,7 @@ function M.locations_to_items(locations)
   local fname = api.nvim_buf_get_name(0)
   for _, d in ipairs(locations) do
     local range = d.range or d.targetSelectionRange
-    table.insert(grouped[fname], {start = range.start})
+    table.insert(grouped[fname], {start = range.start, message = d.message})
   end
 
 
@@ -160,7 +160,7 @@ function M.locations_to_items(locations)
         bufnr = bufnr,
         lnum = row + 1,
         col = col + 1;
-        text = line
+        text = temp.message
       })
     end
   end