diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index c215014..5d581ed 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -4,17 +4,17 @@ set -e
 CONNECT="${CONNECT_ADDR:?CONNECT_ADDR is required}"
 LISTEN="${LISTEN_ADDR:-0.0.0.0:56000}"
 
-VLESS_FLAG=""
-if [ "${VLESS_MODE}" = "true" ]; then
-    VLESS_FLAG="-vless"
+MODE_FLAG=""
+if [ "${MODE}" = "tcp" ]; then
+    MODE_FLAG="-mode tcp"
 fi
 
-WRAP_FLAG=""
-WRAP_KEY_FLAG=""
-if [ "${WRAP_MODE}" = "true" ]; then
-    WRAP="${WRAP_KEY:?WRAP_KEY is required when WRAP_MODE=true}"
-    WRAP_FLAG="-wrap"
-    WRAP_KEY_FLAG="-wrap-key $WRAP"
+OBF_FLAG=""
+OBF_KEY_FLAG=""
+if [ "${OBF_MODE}" = "true" ]; then
+    OBF="${OBF_KEY:?OBF_KEY is required when OBF_MODE=true}"
+    OBF_FLAG="-obf"
+    OBF_KEY_FLAG="-obf-key $OBF"
 fi
 
 DEBUG_FLAG=""
@@ -22,4 +22,4 @@ if [ "${DEBUG}" = "true" ]; then
     DEBUG_FLAG="-debug"
 fi
 
-exec ./server -listen "$LISTEN" -connect "$CONNECT" $VLESS_FLAG $WRAP_FLAG $WRAP_KEY_FLAG $DEBUG_FLAG
+exec ./server -listen "$LISTEN" -connect "$CONNECT" $MODE_FLAG $OBF_FLAG $OBF_KEY_FLAG $DEBUG_FLAG