class RuboCop::Cop::Style::ObjectThen

def check_method_node(node)
  return unless preferred_method(node)
  message = message(node)
  add_offense(node.loc.selector, message: message) do |corrector|
    corrector.replace(node.loc.selector, style.to_s)
  end
end