diff --git a/plugin.sh b/plugin.sh
index ccca798..512fbed 100755
--- a/plugin.sh
+++ b/plugin.sh
@@ -67,9 +67,11 @@ run_stack_deploy() {
 }
 
 run_stack_wait() {
-  docker -H "$REMOTE_DOCKER_HOST" run --rm vitalets/docker-stack-wait-deploy \
+  export DOCKER_HOST="$REMOTE_DOCKER_HOST" 
+  docker run --rm vitalets/docker-stack-wait-deploy \
     | sed 's/True/true/' \
     | bash /dev/stdin "$PLUGIN_STACK"
+  unset DOCKER_HOST
 }
 
 run_purge() {