Fix rubocop crap