recipes/heroku.rb in rails_apps_composer-1.5.3 vs recipes/heroku.rb in rails_apps_composer-1.5.4

@@ -23,11 +23,10 @@
     git :remote => "add staging [e-mail]:#{staging_name}.git"
     say_wizard "Created branches 'production' and 'staging' for Heroku deploy."
   end
 
   unless config['domain'].blank?
-    run "heroku addons:add custom_domains"
-    run "heroku domains:add #{config['domain']}"
+    run "heroku domains:add #{config['domain']} --remote #{config['staging'] ? 'production' : 'heroku'}"
   end
 
   git :push => "#{config['staging'] ? 'staging' : 'heroku'} master" if config['deploy']
 end