diff --git a/main.go b/main.go
index c81cde0..2c409b1 100644
--- a/main.go
+++ b/main.go
@@ -12,8 +12,8 @@ import (
 
 	"github.com/cheetahbyte/flagly/apis"
 	"github.com/cheetahbyte/flagly/internal/audit"
-	"github.com/cheetahbyte/flagly/internal/middleware"
 	"github.com/cheetahbyte/flagly/internal/storage"
+	"github.com/cheetahbyte/flagly/pkg/flagly/middleware"
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"go.uber.org/zap"
diff --git a/internal/middleware/middleware.go b/pkg/flagly/middleware/middleware.go
similarity index 100%
rename from internal/middleware/middleware.go
rename to pkg/flagly/middleware/middleware.go