diff --git a/day01-image/scripts/salt.sh b/day01-image/scripts/salt.sh
index 2869791..d810cb1 100644
--- a/day01-image/scripts/salt.sh
+++ b/day01-image/scripts/salt.sh
@@ -1,5 +1,8 @@
 #!/bin/bash -xe
 
+FORMULA_VERSION=${FORMULA_VERSION:-testing}
+APT_MIRANTIS_SALT_REPO=${APT_MIRANTIS_SALT_REPO:-"http://apt.mirantis.com/xenial/ $FORMULA_VERSION salt"}
+
 echo "deb [arch=amd64] ${APT_MIRANTIS_SALT_REPO}" > /etc/apt/sources.list.d/mcp_salt.list
 apt-get update
 apt-get install salt-formula* -y