diff --git a/blueprint/context.go b/blueprint/context.go
index d43c3af..7a508cb 100644
--- a/blueprint/context.go
+++ b/blueprint/context.go
@@ -1204,8 +1204,7 @@ func (c *Context) generateModuleBuildActions(config interface{},
 						config:  config,
 						group:   group,
 					},
-					module:        module,
-					primaryModule: group.modules[0],
+					module: module,
 				},
 				scope: scope,
 			}
diff --git a/blueprint/module_ctx.go b/blueprint/module_ctx.go
index fd89a86..6faff01 100644
--- a/blueprint/module_ctx.go
+++ b/blueprint/module_ctx.go
@@ -128,6 +128,8 @@ type ModuleContext interface {
 	AddNinjaFileDeps(deps ...string)
 
 	PrimaryModule() Module
+	FinalModule() Module
+	VisitAllModuleVariants(visit func(Module))
 }
 
 var _ BaseModuleContext = (*baseModuleContext)(nil)
@@ -196,8 +198,7 @@ var _ PreModuleContext = (*preModuleContext)(nil)
 
 type preModuleContext struct {
 	baseModuleContext
-	module        *moduleInfo
-	primaryModule *moduleInfo
+	module *moduleInfo
 }
 
 func (m *preModuleContext) OtherModuleName(module Module) string {
@@ -280,7 +281,17 @@ func (m *moduleContext) AddNinjaFileDeps(deps ...string) {
 }
 
 func (m *moduleContext) PrimaryModule() Module {
-	return m.primaryModule.logicModule
+	return m.module.group.modules[0].logicModule
+}
+
+func (m *moduleContext) FinalModule() Module {
+	return m.module.group.modules[len(m.module.group.modules)-1].logicModule
+}
+
+func (m *moduleContext) VisitAllModuleVariants(visit func(Module)) {
+	for _, module := range m.module.group.modules {
+		visit(module.logicModule)
+	}
 }
 
 //