Modify data provider class