lib/capistrano/tasks/migrations.rake in capistrano-rails-1.1.6 vs lib/capistrano/tasks/migrations.rake in capistrano-rails-1.1.7

@@ -2,11 +2,11 @@
 
 namespace :deploy do
 
   desc 'Runs rake db:migrate if migrations are set'
   task :migrate => [:set_rails_env] do
-    on primary fetch(:migration_role) do
+    on fetch(:migration_servers) do
       conditionally_migrate = fetch(:conditionally_migrate)
       info '[deploy:migrate] Checking changes in /db/migrate' if conditionally_migrate
       if conditionally_migrate && test("diff -q #{release_path}/db/migrate #{current_path}/db/migrate")
         info '[deploy:migrate] Skip `deploy:migrate` (nothing changed in db/migrate)'
       else
@@ -16,11 +16,11 @@
     end
   end
 
   desc 'Runs rake db:migrate'
   task migrating: [:set_rails_env] do
-    on primary fetch(:migration_role) do
+    on fetch(:migration_servers) do
       within release_path do
         with rails_env: fetch(:rails_env) do
           execute :rake, 'db:migrate'
         end
       end
@@ -32,7 +32,8 @@
 
 namespace :load do
   task :defaults do
     set :conditionally_migrate, fetch(:conditionally_migrate, false)
     set :migration_role, fetch(:migration_role, :db)
+    set :migration_servers, -> { primary(fetch(:migration_role)) }
   end
 end