1 files changed, 1 insertions, 3 deletions
diff --git a/lib/gitlab/database/load_balancing/setup.rb b/lib/gitlab/database/load_balancing/setup.rb
index ef38f42f50b..126c8bb2aa6 100644
--- a/lib/gitlab/database/load_balancing/setup.rb
+++ b/lib/gitlab/database/load_balancing/setup.rb
@@ -104,11 +104,9 @@ module Gitlab
             end
           end
 
-          # rubocop:disable Database/MultipleDatabases
           def connection
-            use_model_load_balancing? ? super : ActiveRecord::Base.connection
+            use_model_load_balancing? ? super : ApplicationRecord.connection
           end
-          # rubocop:enable Database/MultipleDatabases
         end
       end
     end