diff --git a/omada/docker-compose.yml b/omada/docker-compose.yml
index 5dadb3e..9ae67fe 100644
--- a/omada/docker-compose.yml
+++ b/omada/docker-compose.yml
@@ -1,18 +1,21 @@
 services:
   omada-controller:
-    image: ${IMAGE_NAME:-mbentley/omada-controller:latest}
+    image: ${IMAGE_NAME}
     restart: unless-stopped
+    stop_timeout: ${STOP_TIMEOUT:-60}
     ulimits:
       nofile:
         soft: ${ULIMIT_SOFT:-4096}
-        hard: ${ULIMIT_HARD:-8192}    
+        hard: ${ULIMIT_HARD:-8192}
+    ports:
+      - "${PORT_8088:-8088}:8088"
+      - "${PORT_8043:-8043}:8043"
+      - "${PORT_8843:-8843}:8843"
+      - "${PORT_19810:-19810}:19810/udp"
+      - "${PORT_27001:-27001}:27001/udp"
+      - "${PORT_29810:-29810}:29810/udp"
+      - "${PORT_RANGE_29811_29816:-29811-29816}:29811-29816"
     environment:
-      - MANAGE_HTTP_PORT=${MANAGE_HTTP_PORT:-8088}
-      - MANAGE_HTTPS_PORT=${MANAGE_HTTPS_PORT:-8043}
-      - PORTAL_HTTP_PORT=${PORTAL_HTTP_PORT:-8088}
-      - PORTAL_HTTPS_PORT=${PORTAL_HTTPS_PORT:-8843}
-      - SHOW_SERVER_LOGS=${SHOW_SERVER_LOGS:-true}
-      - SHOW_MONGODB_LOGS=${SHOW_MONGODB_LOGS:-false}
       - TZ=${TZ:-Etc/UTC}
     volumes:
       - ${DATA_VOLUME:-omada-data}:/opt/tplink/EAPController/data