diff --git a/app/app.go b/app/app.go
index 27d596d..81b64df 100644
--- a/app/app.go
+++ b/app/app.go
@@ -12,6 +12,7 @@ import (
 
 type Container struct {
 	DB                  *gorm.DB
+	AgentRepository     agent.Repository
 	RegistrationService *agentService.RegistrationService
 }
 
@@ -24,6 +25,7 @@ func NewContainer(db *gorm.DB) *Container {
 
 	return &Container{
 		DB:                  db,
+		AgentRepository:     agentRepo,
 		RegistrationService: registrationSvc,
 	}
 }
diff --git a/app/middleware/agentauth/auth.go b/app/middleware/agentauth/auth.go
index 04add0a..6d4ed3e 100644
--- a/app/middleware/agentauth/auth.go
+++ b/app/middleware/agentauth/auth.go
@@ -14,6 +14,10 @@ type AgentRepository interface {
 	GetPublicKeyByAgentID(ctx context.Context, agentID string) (string, error)
 }
 
+func New(repo AgentRepository) echo.MiddlewareFunc {
+	return Middleware(repo)
+}
+
 func Middleware(repo AgentRepository) echo.MiddlewareFunc {
 	return func(next echo.HandlerFunc) echo.HandlerFunc {
 		return func(c echo.Context) error {