Fixing rubocop