Fix rubocop