diff --git a/lib/generators/shopify_app/install/templates/shopify_app.rb.tt b/lib/generators/shopify_app/install/templates/shopify_app.rb.tt
index a1ba381b1..9265cd8d8 100644
--- a/lib/generators/shopify_app/install/templates/shopify_app.rb.tt
+++ b/lib/generators/shopify_app/install/templates/shopify_app.rb.tt
@@ -47,7 +47,7 @@ Rails.application.config.after_initialize do
       api_key: ShopifyApp.configuration.api_key,
       api_secret_key: ShopifyApp.configuration.secret,
       api_version: ShopifyApp.configuration.api_version,
-      host: ENV['HOST'],
+      host: ENV.fetch('HOST'),
       scope: ShopifyApp.configuration.scope,
       is_private: !ENV.fetch('SHOPIFY_APP_PRIVATE_SHOP', '').empty?,
       is_embedded: ShopifyApp.configuration.embedded_app,
diff --git a/test/generators/install_generator_test.rb b/test/generators/install_generator_test.rb
index c2224ccc3..248c5ea82 100644
--- a/test/generators/install_generator_test.rb
+++ b/test/generators/install_generator_test.rb
@@ -33,7 +33,7 @@ class InstallGeneratorTest < Rails::Generators::TestCase
               api_key: ShopifyApp.configuration.api_key,
               api_secret_key: ShopifyApp.configuration.secret,
               api_version: ShopifyApp.configuration.api_version,
-              host: ENV['HOST'],
+              host: ENV.fetch('HOST'),
               scope: ShopifyApp.configuration.scope,
               is_private: !ENV.fetch('SHOPIFY_APP_PRIVATE_SHOP', '').empty?,
               is_embedded: ShopifyApp.configuration.embedded_app,