diff --git a/internal/api/admin_handler.go b/internal/api/admin_handler.go
index 7546bee..5cfb140 100644
--- a/internal/api/admin_handler.go
+++ b/internal/api/admin_handler.go
@@ -523,7 +523,7 @@ func (h *AdminHandler) ListKeysForMaster(c *gin.Context) {
 // @Security     AdminAuth
 // @Param        id      path  int  true  "Master ID"
 // @Param        key_id  path  int  true  "Token ID"
-// @Success      200  {object}  ResponseEnvelope{data=dto.DeleteResponse}
+// @Success      200  {object}  ResponseEnvelope{data=MapData}
 // @Failure      400  {object}  ResponseEnvelope{data=MapData}
 // @Failure      404  {object}  ResponseEnvelope{data=MapData}
 // @Failure      500  {object}  ResponseEnvelope{data=MapData}
diff --git a/internal/api/model_registry_handler.go b/internal/api/model_registry_handler.go
index 7525d38..18c68f4 100644
--- a/internal/api/model_registry_handler.go
+++ b/internal/api/model_registry_handler.go
@@ -4,10 +4,17 @@ import (
 	"net/http"
 	"strings"
 
+	"github.com/ez-api/ez-api/internal/dto"
 	"github.com/ez-api/ez-api/internal/service"
 	"github.com/gin-gonic/gin"
 )
 
+// Blank references for swagger doc generation
+var (
+	_ dto.ModelRegistryStatusResponse
+	_ dto.ModelRegistryCheckResponse
+)
+
 type ModelRegistryHandler struct {
 	reg *service.ModelRegistryService
 }