Update to handle new rubocop error added