diff --git a/config.go b/config.go
index aa7869e..6dfb84b 100644
--- a/config.go
+++ b/config.go
@@ -11,6 +11,7 @@ import (
 type Config struct {
 	Enabled        bool   `json:"enabled"`
 	WAFURL         string `json:"waf_url"`
+	ReturnPort     int    `json:"return_port"`
 	HealthInterval int    `json:"health_interval"`
 	TimeoutMs      int    `json:"timeout_ms"`
 }
@@ -32,6 +33,7 @@ func defaultConfig() Config {
 	return Config{
 		Enabled:        false,
 		WAFURL:         "http://127.0.0.1:8080",
+		ReturnPort:     8081,
 		HealthInterval: 5,
 		TimeoutMs:      500,
 	}
diff --git a/main.go b/main.go
index c04568e..90e653f 100644
--- a/main.go
+++ b/main.go
@@ -77,6 +77,9 @@ func handleSaveConfig(w http.ResponseWriter, r *http.Request) {
 	if u := r.FormValue("waf_url"); u != "" {
 		cfg.WAFURL = u
 	}
+	if v := r.FormValue("return_port"); v != "" {
+		fmt.Sscanf(v, "%d", &cfg.ReturnPort)
+	}
 	if v := r.FormValue("health_interval"); v != "" {
 		fmt.Sscanf(v, "%d", &cfg.HealthInterval)
 	}