Adjust to new rubocop rule