lib/capistrano/tasks/migrations.rake in capistrano-rails-1.0.0 vs lib/capistrano/tasks/migrations.rake in capistrano-rails-1.1.0

@@ -1,8 +1,10 @@
+load File.expand_path("../set_rails_env.rake", __FILE__)
+
 namespace :deploy do
 
   desc 'Runs rake db:migrate if migrations are set'
-  task :migrate do
+  task :migrate => [:set_rails_env] do
     on primary fetch(:migration_role) do
       within release_path do
         with rails_env: fetch(:rails_env) do
           execute :rake, "db:migrate"
         end