1 files changed, 5 insertions, 7 deletions
diff --git a/rubocop/cop/include_sidekiq_worker.rb b/rubocop/cop/include_sidekiq_worker.rb
index e39b8bf92c2..1a42de64aa9 100644
--- a/rubocop/cop/include_sidekiq_worker.rb
+++ b/rubocop/cop/include_sidekiq_worker.rb
@@ -3,7 +3,9 @@
 module RuboCop
   module Cop
     # Cop that makes sure workers include `ApplicationWorker`, not `Sidekiq::Worker`.
-    class IncludeSidekiqWorker < RuboCop::Cop::Cop
+    class IncludeSidekiqWorker < RuboCop::Cop::Base
+      extend RuboCop::Cop::AutoCorrector
+
       MSG = 'Include `ApplicationWorker`, not `Sidekiq::Worker`.'
 
       def_node_matcher :includes_sidekiq_worker?, <<~PATTERN
@@ -13,12 +15,8 @@ module RuboCop
       def on_send(node)
         return unless includes_sidekiq_worker?(node)
 
-        add_offense(node.arguments.first, location: :expression)
-      end
-
-      def autocorrect(node)
-        lambda do |corrector|
-          corrector.replace(node.source_range, 'ApplicationWorker')
+        add_offense(node.arguments.first) do |corrector|
+          corrector.replace(node.arguments.first, 'ApplicationWorker')
         end
       end
     end