diff --git a/core/maingate.go b/core/maingate.go
index 3294202..c49e05a 100644
--- a/core/maingate.go
+++ b/core/maingate.go
@@ -431,7 +431,7 @@ func (mg *Maingate) prepare(context context.Context) (err error) {
 
 var portptr = flagx.Int("port", 80, "")
 
-func (mg *Maingate) RegisterHandlers(ctx context.Context, serveMux *http.ServeMux, prefix string) error {
+func (mg *Maingate) RegisterHandlers(ctx context.Context, serveMux gocommon.ServerMuxInterface, prefix string) error {
 	var allServices []*serviceDescription
 	if err := mg.mongoClient.AllAs(CollectionService, &allServices, options.Find().SetReturnKey(false)); err != nil {
 		return logger.ErrorWithCallStack(err)
diff --git a/core/watch.go b/core/watch.go
index 6857dea..6ddd491 100644
--- a/core/watch.go
+++ b/core/watch.go
@@ -3,13 +3,13 @@ package core
 import (
 	"context"
 	"encoding/hex"
-	"net/http"
 	"os"
 	"path"
 	"sync/atomic"
 	"time"
 	"unsafe"
 
+	"repositories.action2quare.com/ayo/gocommon"
 	"repositories.action2quare.com/ayo/gocommon/logger"
 
 	"go.mongodb.org/mongo-driver/bson"
@@ -34,7 +34,7 @@ type filePipelineDocument struct {
 	File *FileDocumentDesc `bson:"fullDocument"`
 }
 
-func (mg *Maingate) watchFileCollection(parentctx context.Context, serveMux *http.ServeMux, prefix string) {
+func (mg *Maingate) watchFileCollection(parentctx context.Context, serveMux gocommon.ServerMuxInterface, prefix string) {
 	defer func() {
 		s := recover()
 		if s != nil {
@@ -119,7 +119,7 @@ func (mg *Maingate) watchFileCollection(parentctx context.Context, serveMux *htt
 	}
 }
 
-func (mg *Maingate) watchServiceCollection(parentctx context.Context, serveMux *http.ServeMux, prefix string) {
+func (mg *Maingate) watchServiceCollection(parentctx context.Context, serveMux gocommon.ServerMuxInterface, prefix string) {
 	defer func() {
 		s := recover()
 		if s != nil {