diff --git a/arm-templates/automate/automate_setup.rb b/arm-templates/automate/automate_setup.rb
index 7619b5a..36eab04 100644
--- a/arm-templates/automate/automate_setup.rb
+++ b/arm-templates/automate/automate_setup.rb
@@ -32,10 +32,14 @@
   config.puts(%Q{api_fqdn "#{@fqdn}"})
 end
 
+environment = {
+  'HOME' => '/root'
+}
+
 # Configure the hostname
-hostname = Mixlib::ShellOut.new("chef-marketplace-ctl hostname #{@fqdn}")
+hostname = Mixlib::ShellOut.new("chef-marketplace-ctl hostname #{@fqdn}", env: environment)
 hostname.run_command
 
 # Configure Automate
-configure = Mixlib::ShellOut.new("chef-marketplace-ctl setup --preconfigure")
+configure = Mixlib::ShellOut.new("chef-marketplace-ctl setup --preconfigure", env: environment)
 configure.run_command
diff --git a/cloudformation/marketplace_byol.yml b/cloudformation/marketplace_byol.yml
index d971ea3..be80214 100644
--- a/cloudformation/marketplace_byol.yml
+++ b/cloudformation/marketplace_byol.yml
@@ -253,6 +253,7 @@ Resources:
           - |+
 
           - - '#!/bin/bash -ex'
+            - export HOME=/root
             - !If
               - HasLicenseUrl
               - !Sub >-