1 files changed, 4 insertions, 3 deletions
diff --git a/lib/gitlab/database/migrations/base_background_runner.rb b/lib/gitlab/database/migrations/base_background_runner.rb
index 76982a9da9b..dbb85bad95c 100644
--- a/lib/gitlab/database/migrations/base_background_runner.rb
+++ b/lib/gitlab/database/migrations/base_background_runner.rb
@@ -4,10 +4,11 @@ module Gitlab
   module Database
     module Migrations
       class BaseBackgroundRunner
-        attr_reader :result_dir
+        attr_reader :result_dir, :connection
 
-        def initialize(result_dir:)
+        def initialize(result_dir:, connection:)
           @result_dir = result_dir
+          @connection = connection
         end
 
         def jobs_by_migration_name
@@ -45,7 +46,7 @@ module Gitlab
 
             instrumentation.observe(version: nil,
                                     name: batch_names.next,
-                                    connection: ActiveRecord::Migration.connection) do
+                                    connection: connection) do
               run_job(j)
             end
           end