diff --git a/docker-compose.yml b/docker-compose.yml
index 036c8dc..1273d4c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,8 +2,8 @@ services:
   dosvault:
     build: .
     ports:
-      - "8080:8080"
-      - "8081:8081"
+      - "${DOSVAULT_PORT:-8080}:8080"
+      - "${DOSVAULT_WEBSOCKET_PORT:-8081}:8081"
     volumes:
       - dosvault_data:/app/data
       - "${ROMS_PATH:-./roms}:/app/data/roms:ro"