1 files changed, 6 insertions, 3 deletions
diff --git a/lib/gitlab/cleanup/orphan_lfs_file_references.rb b/lib/gitlab/cleanup/orphan_lfs_file_references.rb
index a6638b2cbc8..99e7550629a 100644
--- a/lib/gitlab/cleanup/orphan_lfs_file_references.rb
+++ b/lib/gitlab/cleanup/orphan_lfs_file_references.rb
@@ -5,15 +5,14 @@ module Gitlab
     class OrphanLfsFileReferences
       include Gitlab::Utils::StrongMemoize
 
-      attr_reader :project, :dry_run, :logger, :limit
+      attr_reader :project, :dry_run, :logger
 
       DEFAULT_REMOVAL_LIMIT = 1000
 
-      def initialize(project, dry_run: true, logger: nil, limit: nil)
+      def initialize(project, dry_run: true, logger: nil)
         @project = project
         @dry_run = dry_run
         @logger = logger || Gitlab::AppLogger
-        @limit = limit
       end
 
       def run!
@@ -67,6 +66,10 @@ module Gitlab
       def log_info(msg)
         logger.info("#{'[DRY RUN] ' if dry_run}#{msg}")
       end
+
+      def limit
+        ENV['LIMIT']&.to_i
+      end
     end
   end
 end