diff --git a/de/heat-templates/scripts/instance_boot.sh b/de/heat-templates/scripts/instance_boot.sh
index 30881d9..08bb8d7 100644
--- a/de/heat-templates/scripts/instance_boot.sh
+++ b/de/heat-templates/scripts/instance_boot.sh
@@ -191,10 +191,10 @@
     retry $NODE_DEPLOYMENT_RETRIES "The node didn't come up." retry_wait
 }
 
-
 function join_node {
+    local type=${1}
     function retry_join_node {
-        docker swarm join-token $1
+        docker swarm join-token $type
     }
     retry 10 "Failed to join node to swarm" retry_join_node
 }