diff --git a/src/go/ast/filter.go b/src/go/ast/filter.go
index 2be30d2fb4..1f2cab222f 100644
--- a/src/go/ast/filter.go
+++ b/src/go/ast/filter.go
@@ -474,7 +474,8 @@ func MergePackageFiles(pkg *Package, mode MergeMode) *File {
 			}
 		}
 	} else {
-		for _, f := range pkg.Files {
+		for _, filename := range filenames {
+			f := pkg.Files[filename]
 			imports = append(imports, f.Imports...)
 		}
 	}