modify to pass rubocop