diff --git a/docker-compose.prod.yml b/docker-compose.prod.yml
index 711f362..6303f5d 100644
--- a/docker-compose.prod.yml
+++ b/docker-compose.prod.yml
@@ -3,6 +3,11 @@ version: "3.7"
 
 services:
   registry:
+    image: "registry:2"
+    ports:
+      - "5000:5000"
+    volumes:
+      - "registry:/var/lib/registry"
     networks:
       - proxy
     deploy:
@@ -16,8 +21,12 @@ services:
       labels:
         - "traefik.enable=true"
 
-  registry-web-ui:
+  registry-ui:
+    image: "joxit/docker-registry-ui:1.4"
+    ports:
+      - "80:80"
     environment:
+      - REGISTRY_URL=http://registry:5000
       - REGISTRY_TITLE=registry.swarm.autonomic.zone
       - DELETE_IMAGES=true
     networks:
diff --git a/docker-compose.yml b/docker-compose.yml
index 65e9735..e8ddb30 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -9,7 +9,7 @@ services:
     volumes:
       - "registry:/var/lib/registry"
 
-  registry-web-ui:
+  registry-ui:
     image: "joxit/docker-registry-ui:1.4"
     ports:
       - "80:80"