diff --git a/build/ci.go b/build/ci.go
index 2e4f20079b..6a9848876d 100644
--- a/build/ci.go
+++ b/build/ci.go
@@ -322,9 +322,9 @@ func doTest(cmdline []string) {
 		gotest.Args = append(gotest.Args, "-short")
 	}
 
-	packages := []string{"work"}
-	if len(flag.CommandLine.Args()) > 0 {
-		packages = flag.CommandLine.Args()
+	packages := flag.CommandLine.Args()
+	if len(packages) == 0 {
+		packages = workspacePackagePatterns()
 	}
 	gotest.Args = append(gotest.Args, packages...)
 	build.MustRun(gotest)
@@ -364,7 +364,7 @@ func doCheckGenerate() {
 		protocPath      = downloadProtoc(*cachedir)
 		protocGenGoPath = downloadProtocGenGo(*cachedir)
 	)
-	c := tc.Go("generate", "work")
+	c := tc.Go("generate", workspacePackagePatterns()...)
 	pathList := []string{filepath.Join(protocPath, "bin"), protocGenGoPath, os.Getenv("PATH")}
 	c.Env = append(c.Env, "PATH="+strings.Join(pathList, string(os.PathListSeparator)))
 	build.MustRun(c)
@@ -422,30 +422,17 @@ func doCheckBadDeps() {
 func doLint(cmdline []string) {
 	var (
 		cachedir = flag.String("cachedir", "./build/cache", "directory for caching golangci-lint binary.")
-		tc       = new(build.GoToolchain)
 	)
 	flag.CommandLine.Parse(cmdline)
 
 	packages := flag.CommandLine.Args()
 	if len(packages) == 0 {
 		// Get module directories in workspace.
-		listing, err := tc.Go("list", "-m").Output()
-		if err != nil {
-			log.Fatalf("go list failed:", err)
-		}
-		var mainModule []byte
-		for i, m := range bytes.Split(listing, []byte("\n")) {
-			m = bytes.TrimSpace(m)
-			if i == 0 {
-				mainModule = m
-				packages = append(packages, "./...")
-			} else if len(m) > 0 {
-				dir := string(bytes.TrimPrefix(m, mainModule))
-				packages = append(packages, "."+dir+"/...")
-			}
-		}
-		if len(packages) == 0 {
-			log.Fatal("no packages found")
+		packages = []string{"./..."}
+		modules := workspaceModules()
+		for _, m := range modules[1:] {
+			dir := strings.TrimPrefix(m, modules[0])
+			packages = append(packages, "."+dir+"/...")
 		}
 	}
 
@@ -1189,3 +1176,31 @@ func doSanityCheck() {
 	csdb := download.MustLoadChecksums("build/checksums.txt")
 	csdb.DownloadAndVerifyAll()
 }
+
+// workspaceModules lists the module paths in the current work.
+func workspaceModules() []string {
+	listing, err := new(build.GoToolchain).Go("list", "-m").Output()
+	if err != nil {
+		log.Fatalf("go list failed:", err)
+	}
+	var modules []string
+	for _, m := range bytes.Split(listing, []byte("\n")) {
+		m = bytes.TrimSpace(m)
+		if len(m) > 0 {
+			modules = append(modules, string(m))
+		}
+	}
+	if len(modules) == 0 {
+		panic("no modules found")
+	}
+	return modules
+}
+
+func workspacePackagePatterns() []string {
+	modules := workspaceModules()
+	patterns := make([]string, len(modules))
+	for i, m := range modules {
+		patterns[i] = m + "/..."
+	}
+	return patterns
+}