diff --git a/nifi/arm/docker-compose.yml b/nifi/arm/docker-compose.yml
index 0a8aac5..e8657c3 100644
--- a/nifi/arm/docker-compose.yml
+++ b/nifi/arm/docker-compose.yml
@@ -11,8 +11,15 @@ nifi:
     - NIFI_REMOTE_INPUT_SOCKET_PORT=10000
     - NIFI_ZK_CONNECT_STRING=zookeeper1:2181,zookeeper2:2181,zookeeper3:2181
     - NIFI_ELECTION_MAX_WAIT=1 min
+    - NIFI_ELECTION_MAX_CANDIDATES=2
+    - NIFI_JVM_HEAP_INIT=512m
+    - NIFI_JVM_HEAP_MAX=1g
   extra_hosts:
     - zookeeper1:10.0.0.21
     - zookeeper2:10.0.0.22
     - zookeeper3:10.0.0.23
+    # nifi1:10.0.0.11
+    - nifi2:10.0.0.12
+    - nifi3:10.0.0.13
+  hostname: nifi1
   restart: always
diff --git a/nifi/docker-compose.yml b/nifi/docker-compose.yml
index bea32cc..e691647 100644
--- a/nifi/docker-compose.yml
+++ b/nifi/docker-compose.yml
@@ -11,8 +11,15 @@ nifi:
     - NIFI_REMOTE_INPUT_SOCKET_PORT=10000
     - NIFI_ZK_CONNECT_STRING=zookeeper1:2181,zookeeper2:2181,zookeeper3:2181
     - NIFI_ELECTION_MAX_WAIT=1 min
+    - NIFI_ELECTION_MAX_CANDIDATES=2
+    - NIFI_JVM_HEAP_INIT=512m
+    - NIFI_JVM_HEAP_MAX=1g
   extra_hosts:
     - zookeeper1:10.0.0.21
     - zookeeper2:10.0.0.22
     - zookeeper3:10.0.0.23
+    # nifi1:10.0.0.11
+    - nifi2:10.0.0.12
+    - nifi3:10.0.0.13
+  hostname: nifi1
   restart: always