diff --git a/controllers/devbox/cmd/main.go b/controllers/devbox/cmd/main.go
index 62416e9441d..df91e966696 100644
--- a/controllers/devbox/cmd/main.go
+++ b/controllers/devbox/cmd/main.go
@@ -74,6 +74,7 @@ func main() {
 	var requestEphemeralStorage string
 	var limitEphemeralStorage string
 	var debugMode bool
+	var WebSocketImage string
 	flag.StringVar(&registryAddr, "registry-addr", "sealos.hub:5000", "The address of the registry")
 	flag.StringVar(&registryUser, "registry-user", "admin", "The user of the registry")
 	flag.StringVar(&registryPassword, "registry-password", "passw0rd", "The password of the registry")
@@ -93,6 +94,7 @@ func main() {
 	flag.Float64Var(&requestMemoryRate, "request-memory-rate", 10, "The request rate of memory limit in devbox.")
 	flag.StringVar(&requestEphemeralStorage, "request-ephemeral-storage", "500Mi", "The request value of ephemeral storage in devbox.")
 	flag.StringVar(&limitEphemeralStorage, "limit-ephemeral-storage", "10Gi", "The limit value of ephemeral storage in devbox.")
+	flag.StringVar(&WebSocketImage, "websocket-image", "jpillora/chisel", "The image name of devbox websocket proxy pod")
 	opts := zap.Options{
 		Development: true,
 	}