lib/rubocop/cop/rspec/scattered_let.rb in rubocop-rspec-1.41.0 vs lib/rubocop/cop/rspec/scattered_let.rb in rubocop-rspec-1.42.0

@@ -25,36 +25,33 @@
       #     let(:bar) { 2 }
       #     let!(:baz) { 3 }
       #   end
       #
       class ScatteredLet < Cop
+        extend AutoCorrector
+
         MSG = 'Group all let/let! blocks in the example group together.'
 
         def on_block(node)
           return unless example_group_with_body?(node)
 
           check_let_declarations(node.body)
         end
 
-        def autocorrect(node)
-          lambda do |corrector|
-            first_let = find_first_let(node.parent)
-            RuboCop::RSpec::Corrector::MoveNode.new(
-              node, corrector, processed_source
-            ).move_after(first_let)
-          end
-        end
-
         private
 
         def check_let_declarations(body)
           lets = body.each_child_node.select { |node| let?(node) }
 
           first_let = lets.first
           lets.each_with_index do |node, idx|
             next if node.sibling_index == first_let.sibling_index + idx
 
-            add_offense(node)
+            add_offense(node) do |corrector|
+              RuboCop::RSpec::Corrector::MoveNode.new(
+                node, corrector, processed_source
+              ).move_after(first_let)
+            end
           end
         end
 
         def find_first_let(node)
           node.children.find { |child| let?(child) }