diff --git a/cmd/kod/internal/generate_generator.go b/cmd/kod/internal/generate_generator.go
index fb76c70..5d0a7b0 100644
--- a/cmd/kod/internal/generate_generator.go
+++ b/cmd/kod/internal/generate_generator.go
@@ -701,7 +701,6 @@ func (g *generator) generateRegisteredComponents(p printFn) {
 		name := comp.intfName()
 		myName := comp.fullIntfName()
 
-		g.interceptor()
 		localStubFn := fmt.Sprintf(`func(ctx context.Context, info *kod.LocalStubFnInfo) any {
 			return %s_local_stub{
 				impl: info.Impl.(%s),
@@ -765,8 +764,6 @@ func (g *generator) generateLocalStubs(p printFn) {
 	p(``)
 	p(``)
 	p(`// Local stub implementations.`)
-	p(``)
-	g.tset.importPackage("context", "context")
 
 	var b strings.Builder
 	for _, comp := range g.components {
@@ -774,6 +771,9 @@ func (g *generator) generateLocalStubs(p printFn) {
 			continue
 		}
 
+		g.tset.importPackage("context", "context")
+		g.interceptor()
+
 		stub := notExported(comp.intfName()) + "_local_stub"
 		p(`// %s is a local stub implementation of [%s].`, stub, g.tset.genTypeString(comp.intf))
 		p(`type %s struct{`, stub)