diff --git a/backend/cmd/api/main.go b/backend/cmd/api/main.go
index 6e5b6b8..d64b8d9 100644
--- a/backend/cmd/api/main.go
+++ b/backend/cmd/api/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"log"
 
 	"photum-backend/docs"
@@ -110,11 +111,15 @@ func main() {
 	}
 
 	// Swagger UI
-	url := ginSwagger.URL("http://localhost:8080/swagger/doc.json") // The url pointing to API definition
+	swaggerScheme := "http"
+	if cfg.AppEnv == "production" {
+		swaggerScheme = "https"
+	}
+	swaggerURL := fmt.Sprintf("%s://%s/swagger/doc.json", swaggerScheme, cfg.SwaggerHost)
 	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler,
 		ginSwagger.PersistAuthorization(true),
 		ginSwagger.DeepLinking(true),
-		url,
+		ginSwagger.URL(swaggerURL),
 	))
 
 	// Public Routes