Cleaned up Rubocop error