1 files changed, 3 insertions, 3 deletions
diff --git a/lib/gitlab/profiler.rb b/lib/gitlab/profiler.rb
index f8a85f693bc..5af06e82c55 100644
--- a/lib/gitlab/profiler.rb
+++ b/lib/gitlab/profiler.rb
@@ -123,18 +123,18 @@ module Gitlab
 
     def self.with_custom_logger(logger)
       original_colorize_logging = ActiveSupport::LogSubscriber.colorize_logging
-      original_activerecord_logger = ApplicationRecord.logger
+      original_activerecord_logger = ActiveRecord::Base.logger
       original_actioncontroller_logger = ActionController::Base.logger
 
       if logger
         ActiveSupport::LogSubscriber.colorize_logging = false
-        ApplicationRecord.logger = logger
+        ActiveRecord::Base.logger = logger
         ActionController::Base.logger = logger
       end
 
       yield.tap do
         ActiveSupport::LogSubscriber.colorize_logging = original_colorize_logging
-        ApplicationRecord.logger = original_activerecord_logger
+        ActiveRecord::Base.logger = original_activerecord_logger
         ActionController::Base.logger = original_actioncontroller_logger
       end
     end