diff --git a/lua/fzf-lua/providers/lsp.lua b/lua/fzf-lua/providers/lsp.lua
index df0f691..d0b5f53 100644
--- a/lua/fzf-lua/providers/lsp.lua
+++ b/lua/fzf-lua/providers/lsp.lua
@@ -590,7 +590,9 @@ local function get_line_diagnostics(_)
         line = diag[1].end_lnum,
         character = diag[1].end_col,
       }
-    }
+    },
+    data = diag[1].user_data and diag[1].user_data.lsp and
+      diag[1].user_data.lsp.data
   }} or nil
 end