oops missed one

go/tools/gopackagesdriver/packageregistry.go

@@ -58,13 +58,12 @@ func (pr *PackageRegistry) Add(pkgs ...*FlatPackage) *PackageRegistry {
 			Imports:         imports,
 		}
 
-		pr.packagesByID[pkg.ID] = pkg
-
-		// TODO(ellie): wtf why does this fix things
-		if flatPkg.IsStdlib() {
+		if len(pkg.CompiledGoFiles) <= 0 {
 			pkg.CompiledGoFiles = pkg.GoFiles
 		}
 
+		pr.packagesByID[pkg.ID] = pkg
+
 		if flatPkg.IsStdlib() {
 			pr.stdlib[pkg.PkgPath] = pkg
 		}
@@ -116,13 +115,6 @@ func (pr *PackageRegistry) walk(acc map[string]*packages.Package, root string) {
 
 	acc[pkg.ID] = pkg
 	for imp, pkgI := range pkg.Imports {
-		if pkgI == nil {
-			fmt.Fprintf(os.Stderr, "Error: import %s for package %v not resolved\n", imp, root)
-			if _, ok := pr.stdlib[imp]; ok {
-				fmt.Fprintf(os.Stderr, "import was stdlib\n")
-			}
-			continue
-		}
 		if _, ok := acc[pkgI.ID]; !ok {
 			pr.walk(acc, pkgI.ID)
 		}