1 files changed, 2 insertions, 2 deletions
diff --git a/lib/gitlab/database/background_migration/batched_migration.rb b/lib/gitlab/database/background_migration/batched_migration.rb
index 45f52765d0f..92cafd1d00e 100644
--- a/lib/gitlab/database/background_migration/batched_migration.rb
+++ b/lib/gitlab/database/background_migration/batched_migration.rb
@@ -67,11 +67,11 @@ module Gitlab
           end
 
           event :finish do
-            transition any => :finished
+            transition [:paused, :finished, :active, :finalizing] => :finished
           end
 
           event :failure do
-            transition any => :failed
+            transition [:failed, :finalizing, :active] => :failed
           end
 
           event :finalize do