diff --git a/src/Module.zig b/src/Module.zig
index 558c03d685..9d3810f671 100644
--- a/src/Module.zig
+++ b/src/Module.zig
@@ -5307,7 +5307,7 @@ pub fn populateTestFunctions(
 ) !void {
     const gpa = mod.gpa;
     const ip = &mod.intern_pool;
-    const builtin_mod = mod.main_mod.getBuiltinDependency();
+    const builtin_mod = mod.root_mod.getBuiltinDependency();
     const builtin_file = (mod.importPkg(builtin_mod) catch unreachable).file;
     const root_decl = mod.declPtr(builtin_file.root_decl.unwrap().?);
     const builtin_namespace = mod.namespacePtr(root_decl.src_namespace);
diff --git a/src/Package/Module.zig b/src/Package/Module.zig
index 3c3a1c81e6..66d5a21eab 100644
--- a/src/Package/Module.zig
+++ b/src/Package/Module.zig
@@ -466,6 +466,9 @@ pub fn createLimited(gpa: Allocator, options: LimitedOptions) Allocator.Error!*P
     return mod;
 }
 
+/// Asserts that the module has a builtin module, which is not true for non-zig
+/// modules such as ones only used for `@embedFile`, or the root module when
+/// there is no Zig Compilation Unit.
 pub fn getBuiltinDependency(m: Module) *Module {
     const result = m.deps.values()[0];
     assert(result.isBuiltin());