diff --git a/generators/do-swarm-workers/templates/do-swarm-workers/Pulumi.stack.yaml b/generators/do-swarm-workers/templates/do-swarm-workers/Pulumi.stack.yaml
index 8e29c4c..717e823 100644
--- a/generators/do-swarm-workers/templates/do-swarm-workers/Pulumi.stack.yaml
+++ b/generators/do-swarm-workers/templates/do-swarm-workers/Pulumi.stack.yaml
@@ -1,6 +1,5 @@
 config:
-  aws:region: ap-southeast-1
-  <%= projectName %>:count: "1"
+#  <%= projectName %>:count: "1"
   <%= projectName %>:image: docker-20-04
   <%= projectName %>:name: <%= prefix %>-<%= environment %>
   <%= projectName %>:pathToSshKeysFolder: "../../ssh-keys"
@@ -17,4 +16,4 @@ config:
   <%= projectName %>:userGroups: group1,group2
   <%= projectName %>:username: USERNAME
 #  <%= projectName %>:vpcUuid: VPC UUID
-#  <%= projectName %>:vpcStack: <%= prefix %>-do-resources
\ No newline at end of file
+#  <%= projectName %>:vpcStack: <%= prefix %>-do-resources
diff --git a/generators/do-swarm-workers/templates/do-swarm-workers/cloud-config.njx b/generators/do-swarm-workers/templates/do-swarm-workers/cloud-config.njx
index 690b5d8..654ecd9 100644
--- a/generators/do-swarm-workers/templates/do-swarm-workers/cloud-config.njx
+++ b/generators/do-swarm-workers/templates/do-swarm-workers/cloud-config.njx
@@ -2,7 +2,7 @@
 users:
 {% for user in users %}
   - name: {{ user.username }}
-    groups: {{ user.groups }}
+    groups: docker{% if user.groups %},{{ user.groups }}{% endif %}
     shell: /bin/bash
     sudo: ['ALL=(ALL) NOPASSWD:ALL']
     ssh-authorized-keys: 
@@ -47,4 +47,4 @@ runcmd:
 {% for volume
  in volumes %}
   - chown -R {{ volume.user }}:{{ volume.group }} {{ volume.path }}
-{% endfor %}
\ No newline at end of file
+{% endfor %}
diff --git a/generators/do-swarm-workers/templates/do-swarm-workers/config.ts b/generators/do-swarm-workers/templates/do-swarm-workers/config.ts
index a7987f8..2b4567e 100644
--- a/generators/do-swarm-workers/templates/do-swarm-workers/config.ts
+++ b/generators/do-swarm-workers/templates/do-swarm-workers/config.ts
@@ -54,7 +54,7 @@ export const getConfig = async () => {
   }
 
   return {
-    count: stackConfig.requireNumber("count"),
+    count: stackConfig.getNumber("count") || 1,
     image: stackConfig.require("image"),
     name: stackConfig.get("name") || stack,
     packages,
diff --git a/generators/do-swarm-workers/templates/do-swarm-workers/index.ts b/generators/do-swarm-workers/templates/do-swarm-workers/index.ts
index 0071f4e..a891212 100644
--- a/generators/do-swarm-workers/templates/do-swarm-workers/index.ts
+++ b/generators/do-swarm-workers/templates/do-swarm-workers/index.ts
@@ -12,7 +12,7 @@ export = async () => {
   };
 
   // Desired worker count
-  const count = config.count || 1;
+  const count = config.count;
 
   const workers = [];