lib/capistrano/tasks/set_rails_env.rake in capistrano-rails-1.1.1 vs lib/capistrano/tasks/set_rails_env.rake in capistrano-rails-1.1.2

@@ -1,5 +1,9 @@
 namespace :deploy do
   task :set_rails_env do
     set :rails_env, (fetch(:rails_env) || fetch(:stage))
   end
-end
\ No newline at end of file
+end
+
+Capistrano::DSL.stages.each do |stage|
+  after stage, 'deploy:set_rails_env'
+end