refactor setup_class to provider