Rubocop
Fix code LOL