1 files changed, 2 insertions, 2 deletions
diff --git a/app/services/packages/create_dependency_service.rb b/app/services/packages/create_dependency_service.rb
index 2999885d55d..2c80ec66dbc 100644
--- a/app/services/packages/create_dependency_service.rb
+++ b/app/services/packages/create_dependency_service.rb
@@ -27,7 +27,7 @@ module Packages
         dependencies_to_insert = names_and_version_patterns.reject { |k, _| k.in?(existing_names) }
       end
 
-      ActiveRecord::Base.transaction do
+      ApplicationRecord.transaction do
         inserted_ids = bulk_insert_package_dependencies(dependencies_to_insert)
         bulk_insert_package_dependency_links(type, (existing_ids + inserted_ids))
       end
@@ -76,7 +76,7 @@ module Packages
     end
 
     def database
-      ::Gitlab::Database
+      ::Gitlab::Database.main
     end
   end
 end