Made bust test to use a data provider