Fix rubocop regression