go-kod
diff --git a/‎cmd/kod/internal/generate_generator.go
+5-1 b/‎cmd/kod/internal/generate_generator.go
+5-1
diff --git a/‎examples/helloworld/kod_gen.go
+1-7 b/‎examples/helloworld/kod_gen.go
+1-7
diff --git a/‎tests/case1/kod_gen.go
+5-11 b/‎tests/case1/kod_gen.go
+5-11
diff --git a/‎tests/case2/kod_gen.go
+5-11 b/‎tests/case2/kod_gen.go
+5-11
diff --git a/‎tests/case3/kod_gen.go
+1-7 b/‎tests/case3/kod_gen.go
+1-7
diff --git a/‎tests/case4/kod_gen.go
+1-7 b/‎tests/case4/kod_gen.go
+1-7
diff --git a/‎tests/case5/kod_gen.go
+5-11 b/‎tests/case5/kod_gen.go
+5-11
diff --git a/‎tests/graphcase/kod_gen.go
+1-7 b/‎tests/graphcase/kod_gen.go
+1-7
@@ -720,7 +720,11 @@ func (g *generator) generateRegisteredComponents(p printFn) {
 		p(`		Interface: %s((*%s)(nil)).Elem(),`, reflect.qualify("TypeOf"), g.componentRef(comp))
 		p(`		Impl: %s(%s{}),`, reflect.qualify("TypeOf"), comp.implName())
 		p("		Refs: `%s`,", strings.Join(refNames, ",\n"))
-		p(`		LocalStubFn: %s,`, localStubFn)
+		if !comp.isMain {
+			p(`		LocalStubFn: %s,`, localStubFn)
+		} else {
+			p(`		LocalStubFn: nil,`)
+		}
 		p(`	})`)
 	}
 	p(`}`)