diff --git a/go/private/context.bzl b/go/private/context.bzl
index a52f6e4f8..17817f86f 100644
--- a/go/private/context.bzl
+++ b/go/private/context.bzl
@@ -309,7 +309,7 @@ def new_source(
     for k, v in getattr(attr, "x_defs", {}).items():
         v = _expand_location(go, attr, v)
         if "." not in k:
-            k = "{}.{}".format(source["importmap"], k)
+            k = "{}.{}".format(importmap, k)
         x_defs[k] = v
     source["x_defs"] = x_defs
     if not source["cgo"]:
diff --git a/go/private/rules/nogo.bzl b/go/private/rules/nogo.bzl
index ec6d13994..288a02912 100644
--- a/go/private/rules/nogo.bzl
+++ b/go/private/rules/nogo.bzl
@@ -65,10 +65,10 @@ def _nogo_impl(ctx):
     nogo_source = new_source(
         go,
         struct(
-            srcs = [struct(files = [nogo_main])],
             embed = [ctx.attr._nogo_srcs],
             deps = analyzer_archives,
         ),
+        generated_srcs = [nogo_main],
         name = go.label.name + "~nogo",
         importpath = "nogomain",
         pathtype = EXPORT_PATH,
diff --git a/go/private/rules/test.bzl b/go/private/rules/test.bzl
index 25eca2aa9..31ba3cedc 100644
--- a/go/private/rules/test.bzl
+++ b/go/private/rules/test.bzl
@@ -147,13 +147,13 @@ def _go_test_impl(ctx):
     test_source = new_source(
         go,
         struct(
-            srcs = [struct(files = [main_go])],
             deps = test_deps,
         ),
         name = go.label.name + "~testmain",
         importpath = "testmain",
         pathtype = INFERRED_PATH,
         is_main = True,
+        generated_srcs = [main_go],
         coverage_instrumented = False,
     )
     test_archive, executable, runfiles = go.binary(