Fix Rubocop warnings