From 5da6048c91ca10ef77115974be177d4e11db40db Mon Sep 17 00:00:00 2001
From: Michael Smith <mike@spokefire.co.uk>
Date: Thu, 14 Mar 2013 11:38:11 +0000
Subject: [PATCH] Fixes to generated config file and deploy directory

---
 lib/capistrano/tasks/install.rake     | 2 +-
 lib/capistrano/templates/stage.rb.erb | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/capistrano/tasks/install.rake b/lib/capistrano/tasks/install.rake
index 2cfb485c..94d26edc 100644
--- a/lib/capistrano/tasks/install.rake
+++ b/lib/capistrano/tasks/install.rake
@@ -11,7 +11,7 @@ task :install do
   stage_rb = File.expand_path("../../templates/stage.rb.erb", __FILE__)
   capfile = File.expand_path("../../templates/Capfile", __FILE__)
 
-  mkdir_p config_dir
+  mkdir_p deploy_dir
 
   template = File.read(deploy_rb)
   file = File.join(config_dir, 'deploy.rb')
diff --git a/lib/capistrano/templates/stage.rb.erb b/lib/capistrano/templates/stage.rb.erb
index 59759930..427f109f 100644
--- a/lib/capistrano/templates/stage.rb.erb
+++ b/lib/capistrano/templates/stage.rb.erb
@@ -1,4 +1,5 @@
-stage :<%= stage %>
+set :stage, :<%= stage %>
+
 role :app, %w{example.com}
 role :web, %w{example.com}
 role :db, %w{example.com}