tidy up for rubocop