modify as rubocop suggested