diff --git a/pkg/extension/service/extension.go b/pkg/extension/service/extension.go
index a92d16235642f9b63c7960952414bb427c985eb0..a89eba65db08ce25f455dfb15e3f803e24af79de 100644
--- a/pkg/extension/service/extension.go
+++ b/pkg/extension/service/extension.go
@@ -38,12 +38,13 @@ type Extension struct {
 	keyFn      extension.KeyFunc
 }
 
-func NewExtension(desc *extension.ExtensionDescriptor, cnt *content.Content, setupFunc SetupFunc, logger *zap.Logger) *Extension {
+func NewExtension(desc *extension.ExtensionDescriptor, cnt *content.Content, setupFunc SetupFunc, router extension.ActionRouter, logger *zap.Logger) *Extension {
 	if logger == nil {
 		logger = zap.NewNop()
 	}
 	return &Extension{
 		desc:      desc,
+		router:    router,
 		setupFunc: setupFunc,
 		Content:   cnt,
 		Logger:    logger,