diff --git a/internal/lsp/mod/code_lens.go b/internal/lsp/mod/code_lens.go
index 1598ed5..d7310c7 100644
--- a/internal/lsp/mod/code_lens.go
+++ b/internal/lsp/mod/code_lens.go
@@ -50,7 +50,7 @@
 	upgradeTransitive, err := command.NewUpgradeDependencyCommand("Upgrade transitive dependencies", command.DependencyArgs{
 		URI:        uri,
 		AddRequire: false,
-		GoCmdArgs:  []string{"-u", "all"},
+		GoCmdArgs:  []string{"-d", "-u", "-t", "./..."},
 	})
 	if err != nil {
 		return nil, err
@@ -58,7 +58,7 @@
 	upgradeDirect, err := command.NewUpgradeDependencyCommand("Upgrade direct dependencies", command.DependencyArgs{
 		URI:        uri,
 		AddRequire: false,
-		GoCmdArgs:  requires,
+		GoCmdArgs:  append([]string{"-d"}, requires...),
 	})
 	if err != nil {
 		return nil, err