diff --git a/docker-compose.yaml b/docker-compose.yaml
index 0f172eb..a120e85 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -11,7 +11,8 @@ services:
     volumes:
     - ./experiments:/pyrate/experiments
     - ./data:/pyrate/data
-    scale: 0
+    deploy:
+      replicas: 1
 
   pyrate-gen:
     build:
@@ -24,7 +25,8 @@ services:
     - save_frequency=50
     - seed_start=150
     - continues=False
-    scale: 8
+    deploy:
+      replicas: 4
 
 volumes:
   experiments: