diff --git a/gopls/doc/vim.md b/gopls/doc/vim.md
index 48c9b03d59..a6b40a46b3 100644
--- a/gopls/doc/vim.md
+++ b/gopls/doc/vim.md
@@ -165,7 +165,7 @@ lua <<EOF
   -- …
 
   function goimports(timeout_ms)
-    local context = { source = { organizeImports = true } }
+    local context = { only = { "source.organizeImports" } }
     vim.validate { context = { context, "t", true } }
 
     local params = vim.lsp.util.make_range_params()