Fix rubocop warnings